[Mlir-commits] [mlir] Add MemoryEffect to xegpu.atomic_rmw op (PR #128076)

[Nishant Patel]

https://github.com/nbpatel created https://github.com/llvm/llvm-project/pull/128076

None

>From 4b69517011f377cb06a61a1699fc2616319ab180 Mon Sep 17 00:00:00 2001
From: nbpatel <nishant.b.patel at intel.com>
Date: Thu, 20 Feb 2025 21:55:01 +0000
Subject: [PATCH] Add MemoryEffect to xegpu.atomic_rmw op

---
 mlir/include/mlir/Dialect/XeGPU/IR/XeGPUOps.td | 1 +
 1 file changed, 1 insertion(+)

diff --git a/mlir/include/mlir/Dialect/XeGPU/IR/XeGPUOps.td b/mlir/include/mlir/Dialect/XeGPU/IR/XeGPUOps.td
index 7560ede058faa..881f18dbed29c 100644
--- a/mlir/include/mlir/Dialect/XeGPU/IR/XeGPUOps.td
+++ b/mlir/include/mlir/Dialect/XeGPU/IR/XeGPUOps.td
@@ -788,6 +788,7 @@ def XeGPU_DpasOp : XeGPU_Op<"dpas", [Pure, AllElementTypesMatch<["lhs", "rhs"]>]
 }
 
 def XeGPU_AtomicRMWOp: XeGPU_Op<"atomic_rmw", [Pure,
+      MemoryEffects<[MemWrite]>,
       AllElementTypesMatch<["tensorDesc", "value", "result"]>,
       AllShapesMatch<["tensorDesc", "value", "result"]>]> {
   let summary = "Atomic ready-modify-write operation on the TensorDesc. ";