Skip to content

Commit 37c656a

Browse files
mwhittakerGoogle-ML-Automation
authored andcommitted
Add stablehlo.async_start and stablehlo.async_done.
See openxla/stablehlo#2897 for context. This CL introduces a `stablehlo.future` type and `stablehlo.async_start` and `stablehlo.async_done`. It does not add any translation for them. That will come in a later change. I also did not make `async_start` and `async_done` variadic for now, even though some collectives are variadic. We can add support for that later if we need it. PiperOrigin-RevId: 890017379
1 parent 31bf56f commit 37c656a

File tree

4 files changed

+4612
-0
lines changed

4 files changed

+4612
-0
lines changed

0 commit comments

Comments
 (0)