[Mlir-commits] [mlir] 73dec82 - [ROCDL] Hot fix of mem trait in l2 prefetch (#183550)

[llvmlistbot at llvm.org]

Author: Ravil Dorozhinskii
Date: 2026-02-26T16:50:26+01:00
New Revision: 73dec8293b8624668965c5fcf80baa9456fbd0c6

URL: https://github.com/llvm/llvm-project/commit/73dec8293b8624668965c5fcf80baa9456fbd0c6
DIFF: https://github.com/llvm/llvm-project/commit/73dec8293b8624668965c5fcf80baa9456fbd0c6.diff

LOG: [ROCDL] Hot fix of mem trait in l2 prefetch (#183550)

Added: 
    

Modified: 
    mlir/include/mlir/Dialect/LLVMIR/ROCDLOps.td

Removed: 
    


################################################################################
diff  --git a/mlir/include/mlir/Dialect/LLVMIR/ROCDLOps.td b/mlir/include/mlir/Dialect/LLVMIR/ROCDLOps.td
index be52d1ed9e17c..1f2d472611120 100644
--- a/mlir/include/mlir/Dialect/LLVMIR/ROCDLOps.td
+++ b/mlir/include/mlir/Dialect/LLVMIR/ROCDLOps.td
@@ -2092,7 +2092,7 @@ def ROCDL_RawBufferAtomicCmpSwap :
 
 def ROCDL_GlobalPrefetchOp :
   ROCDL_IntrOp<"global.prefetch", [], [], [], 0, 0, 1, 0, [1], ["scope"]> {
-  dag args = (ins Arg<LLVM_PointerInAddressSpace<1>, "", [MemRead, MemRead]>:$ptr,
+  dag args = (ins Arg<LLVM_PointerInAddressSpace<1>, "", [MemWrite, MemRead]>:$ptr,
                   I32Attr:$scope);
   let arguments = !con(args, baseArgs);
   let description = [{
@@ -2116,7 +2116,7 @@ def ROCDL_GlobalPrefetchOp :
 
 def ROCDL_FlatPrefetchOp :
   ROCDL_IntrOp<"flat.prefetch", [], [], [], 0, 0, 1, 0, [1], ["scope"]> {
-  dag args = (ins Arg<LLVM_PointerInAddressSpace<0>, "", [MemRead, MemRead]>:$ptr,
+  dag args = (ins Arg<LLVM_PointerInAddressSpace<0>, "", [MemWrite, MemRead]>:$ptr,
                   I32Attr:$scope);
   let arguments = !con(args, baseArgs);
   let description = [{