Skip to content

Commit 9546afe

Browse files
authored
[MLIR][XeGPU]Add MemoryEffect to xegpu.atomic_rmw op (#128076)
1 parent 0c50054 commit 9546afe

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

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

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -788,9 +788,10 @@ def XeGPU_DpasOp : XeGPU_Op<"dpas", [Pure, AllElementTypesMatch<["lhs", "rhs"]>]
788788
}
789789

790790
def XeGPU_AtomicRMWOp: XeGPU_Op<"atomic_rmw", [Pure,
791+
MemoryEffects<[MemRead, MemWrite]>,
791792
AllElementTypesMatch<["tensorDesc", "value", "result"]>,
792793
AllShapesMatch<["tensorDesc", "value", "result"]>]> {
793-
let summary = "Atomic ready-modify-write operation on the TensorDesc. ";
794+
let summary = "Atomic read-modify-write operation on the TensorDesc. ";
794795

795796
let description = [{
796797
The `xegpu.atomic_rmw` operation provides a way to perform a read-modify-write

0 commit comments

Comments
 (0)