Skip to content

Commit 2a5f8d3

Browse files
mwhittakertensorflower-gardener
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: 874740969
1 parent b68523c commit 2a5f8d3

4 files changed

Lines changed: 4612 additions & 0 deletions

File tree

0 commit comments

Comments
 (0)