[Mlir-commits] [mlir] [MLIR] Add ReturnLike trait to memref.atomic_yield (PR #144932)
Oleksandr Alex Zinenko
llvmlistbot at llvm.org
Thu Jun 19 11:13:44 PDT 2025
================
@@ -510,3 +510,18 @@ module {
// CHECK: %[[yield:.*]] = arith.addf %{{.*}}, %{{.*}} : f32
// CHECK: linalg.yield %[[yield]] : f32
// CHECK-NOT: arith.subf
+
+// -----
+
+// CHECK-LABEL: func.func @test_atomic_yield
+func.func @test_atomic_yield(%I: memref<10xf32>, %idx : index) {
+ // CHECK: memref.generic_atomic_rmw
+ %x = memref.generic_atomic_rmw %I[%idx] : memref<10xf32> {
+ ^bb0(%current_value : f32):
+ // CHECK: arith.constant
+ %c1 = arith.constant 1.0 : f32
+ // CHECK: memref.atomic_yield
----------------
ftynse wrote:
```suggestion
// CHECK: arith.constant
%c1 = arith.constant 1.0 : f32
// CHECK: memref.atomic_yield
```
https://github.com/llvm/llvm-project/pull/144932
More information about the Mlir-commits
mailing list