Commit 2a5f8d3
Add
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: 874740969stablehlo.async_start and stablehlo.async_done.1 parent b68523c commit 2a5f8d3
4 files changed
Lines changed: 4612 additions & 0 deletions
File tree
- third_party/xla
- third_party/stablehlo
- xla
- hlo/translate/mhlo_to_hlo
- mlir_hlo/mhlo/transforms
0 commit comments