Skip to content

Commit 31b2b23

Browse files
authored
[TritonNVIDIAGPU] Add MemRead<GlobalMemory> to async TMA write ops (triton-lang#6485)
1 parent f8a19d1 commit 31b2b23

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

include/triton/Dialect/TritonNvidiaGPU/IR/TritonNvidiaGPUOps.td

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -315,7 +315,7 @@ def TTNG_AsyncTMACopyLocalToGlobalOp : TTNG_Op<"async_tma_copy_local_to_global">
315315
}];
316316

317317
let arguments = (ins
318-
Arg<TT_PtrType, "", [MemWrite<GlobalMemory>]>:$desc_ptr,
318+
Arg<TT_PtrType, "", [MemRead<GlobalMemory>, MemWrite<GlobalMemory>]>:$desc_ptr,
319319
Variadic<I32>:$coord,
320320
Arg<TTG_MemDescType, "", [MemRead<SharedMemory>]>:$src
321321
);
@@ -364,7 +364,7 @@ def TTNG_AsyncTMAScatterOp : TTNG_Op<"async_tma_scatter"> {
364364
}];
365365

366366
let arguments = (ins
367-
Arg<TT_PtrType ,"", [MemWrite<GlobalMemory>]>:$desc_ptr,
367+
Arg<TT_PtrType ,"", [MemRead<GlobalMemory>, MemWrite<GlobalMemory>]>:$desc_ptr,
368368
RankedTensorOf<[I32]>:$x_offsets,
369369
I32:$y_offset,
370370
Arg<TTG_MemDescType, "", [MemRead<SharedMemory>]>:$src

0 commit comments

Comments
 (0)