mirror-llvm-project/mlir/test/Dialect/GPU
Ivan R. Ivanov 056dae16f7
[mlir][gpu] Add stream-based async mode to gpu.launch (#213031)
Currently, only gpu.launch_func supports stream-based asynchronicity. This
patch adds the same mode to gpu.launch, and makes kernel outlining preserve
that when converting to gpu.launch_func.

This patch also documents the stream-based model in the operations description,
which was previously missing.

This patch changes the builders for the two launch operations:

To retain the current behavior for gpu.launch one needs to add a `nullptr` for
the new `Value asyncObject` argument after the `ValueRange asyncDependencies`,
and make sure to pass a valid type for the `Type asyncTokenType` parameter if
dependency-based async behavior is intended.

For gpu.launch_func, both stream-based and dependency-based async models now go
through the same builders, which take all three `Type asyncTokenType,
ValueRange asyncDependencies, Value asyncObject` arguments. To retain
stream-based behavior, pass `nullptr, {}` to the first two parameters, while to
retain dependency-based behavior, pass `nullptr` to the third parameter.

Assisted-by: Claude Code
2026-07-30 19:30:51 +02:00
..
all-reduce-add.mlir [mlir] Update all-reduce (& vector tests) to use workgroup barriers (#178285) 2026-01-27 13:09:18 -08:00
all-reduce-maxf.mlir [mlir] Update all-reduce (& vector tests) to use workgroup barriers (#178285) 2026-01-27 13:09:18 -08:00
async-region.mlir
barrier-elimination.mlir [mlir][GPU] Extend gpu.barrier with scope and named-barrier support (#195692) 2026-05-18 10:12:34 -07:00
broadcast-speculatability.mlir [mlir][gpu] Add gpu.ballot operation to GPU dialect (#188647) 2026-03-27 11:48:47 -04:00
bufferization-buffer-deallocation.mlir
canonicalize.mlir [mlir][gpu] Fix memref.dim folding with negative index (#205338) 2026-06-24 09:24:43 +08:00
constant-address-space.mlir [mlir][GPU] Add constant address space to GPU dialect (#190211) 2026-04-02 15:02:12 -04:00
decompose-memrefs.mlir
dynamic-shared-memory.mlir
fold-memref-alias-ops.mlir [mlir][MemRef][GPU] Migrate GPU dialect ops to IndexedAccessOpInterface (#190380) 2026-04-20 10:19:57 -07:00
globalId-rewrite.mlir [mlir][gpu] Use arith dialect to lower gpu.global_id (#171614) 2025-12-13 18:43:12 +08:00
indirect-device-func-call.mlir
int-range-interface-cluster.mlir [mlir][GPU] Bump static bound on cluster IDs (#187106) 2026-03-18 11:33:20 -07:00
int-range-interface.mlir
invalid.mlir [mlir][gpu] Add stream-based async mode to gpu.launch (#213031) 2026-07-30 19:30:51 +02:00
mapping.mlir
memref-to-llvm.mlir
module-to-binary-invalid-format.mlir [MLIR] Fix invalid test after improving the error message (NFC) 2026-02-28 15:25:43 -08:00
module-to-binary-invalid.mlir
module-to-binary-nvvm.mlir
module-to-binary-rocdl.mlir [mlir][LLVM][ROCDL] Add buffer oob mode module flags, flag interfaces (#202729) 2026-06-12 09:19:39 -07:00
module-to-binary-spirv.mlir
multiple-all-reduce.mlir
named-barrier.mlir [mlir][GPU] Extend gpu.barrier with scope and named-barrier support (#195692) 2026-05-18 10:12:34 -07:00
nvvm-attach-target.mlir [MLIR][GPU] Update serializeToObject to use SerializedObject wrapper and include ISA compiler logs (#176697) 2026-01-30 12:56:20 +01:00
ops.mlir [mlir][gpu] Add stream-based async mode to gpu.launch (#213031) 2026-07-30 19:30:51 +02:00
outlining.mlir [MLIR][GPU] Add cooperative launch support to gpu.launch_func (#190639) 2026-04-29 11:33:22 +02:00
promote-shuffle-amdgpu.mlir
promotion.mlir
shuffle-rewrite.mlir [mlir][ODS] Fix notorious double-space bug in op printers (#184253) 2026-03-06 16:32:01 -05:00
sink-ops.mlir
sparse-roundtrip.mlir [mlir][ODS] Fix notorious double-space bug in op printers (#184253) 2026-03-06 16:32:01 -05:00
spirv-attach-targets.mlir
subgroup-mma-vector-unroll.mlir [mlir][GPU] Fix double spaces in tests after ODS printer fix. NFC. (#185325) 2026-03-08 18:46:54 -04:00
subgroup-reduce-lowering.mlir [MLIR][GPU] Ensure all lanes in cluster have final reduction value (#165764) 2025-10-31 09:12:43 -07:00
subgroupId-rewrite.mlir [mlir][GPU] Set nsw/nuw when expanding out subgroup ID (#187099) 2026-03-17 19:32:19 +00:00
test-nvvm-pipeline.mlir
transform-gpu-failing.mlir [mlir][gpu] Harden verification constraints in transform.gpu.map_nested_forall_to_threads (#170282) 2026-02-18 08:41:45 +09:00
transform-gpu.mlir [mlir][GPU] Fix double spaces in tests after ODS printer fix. NFC. (#185325) 2026-03-08 18:46:54 -04:00
value-bounds-op-interface-impl.mlir [mlir][GPU] Add ValueBoundsOphinterface to gpu.subgroup_broadcast (#183848) 2026-02-27 15:46:56 -08:00