[Mlir-commits] [mlir] [MLIR] Add ReturnLike trait to memref.atomic_yield (PR #144932)

[Oleksandr Alex Zinenko]

================
@@ -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