[Mlir-commits] [mlir] 9546afe - [MLIR][XeGPU]Add MemoryEffect to xegpu.atomic_rmw op (#128076)

[llvmlistbot at llvm.org]

Author: Nishant Patel
Date: 2025-02-21T11:23:34-06:00
New Revision: 9546afe86c69c2a9421898025a5443e51f00969c

URL: https://github.com/llvm/llvm-project/commit/9546afe86c69c2a9421898025a5443e51f00969c
DIFF: https://github.com/llvm/llvm-project/commit/9546afe86c69c2a9421898025a5443e51f00969c.diff

LOG: [MLIR][XeGPU]Add MemoryEffect to xegpu.atomic_rmw op (#128076)

Added: 
    

Modified: 
    mlir/include/mlir/Dialect/XeGPU/IR/XeGPUOps.td

Removed: 
    


################################################################################
diff  --git a/mlir/include/mlir/Dialect/XeGPU/IR/XeGPUOps.td b/mlir/include/mlir/Dialect/XeGPU/IR/XeGPUOps.td
index 7560ede058faa..78dfaef97420b 100644
--- a/mlir/include/mlir/Dialect/XeGPU/IR/XeGPUOps.td
+++ b/mlir/include/mlir/Dialect/XeGPU/IR/XeGPUOps.td
@@ -788,9 +788,10 @@ def XeGPU_DpasOp : XeGPU_Op<"dpas", [Pure, AllElementTypesMatch<["lhs", "rhs"]>]
 }
 
 def XeGPU_AtomicRMWOp: XeGPU_Op<"atomic_rmw", [Pure,
+      MemoryEffects<[MemRead, MemWrite]>,
       AllElementTypesMatch<["tensorDesc", "value", "result"]>,
       AllShapesMatch<["tensorDesc", "value", "result"]>]> {
-  let summary = "Atomic ready-modify-write operation on the TensorDesc. ";
+  let summary = "Atomic read-modify-write operation on the TensorDesc. ";
 
   let description = [{
     The `xegpu.atomic_rmw` operation provides a way to perform a read-modify-write