Skip to content

Commit ed6f1b7

Browse files
authored
[BACKEND] CLCTryCancel uses the async proxy (#9464)
As per title
1 parent 16da670 commit ed6f1b7

1 file changed

Lines changed: 5 additions & 1 deletion

File tree

lib/Dialect/TritonNvidiaGPU/Transforms/ProxyFenceInsertion.cpp

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,8 @@ namespace {
3232

3333
bool isAsyncProxyWrite(Operation *op) {
3434
return isa<triton::nvidia_gpu::AsyncTMACopyGlobalToLocalOp,
35-
triton::nvidia_gpu::AsyncTMAGatherOp>(op);
35+
triton::nvidia_gpu::AsyncTMAGatherOp,
36+
triton::nvidia_gpu::CLCTryCancelOp>(op);
3637
}
3738

3839
Value getSmemDest(Operation *op) {
@@ -44,6 +45,9 @@ Value getSmemDest(Operation *op) {
4445
dyn_cast<triton::nvidia_gpu::AsyncTMAGatherOp>(op)) {
4546
return asyncTMAGatherOp.getResult();
4647
}
48+
if (auto clcTryCancelOp = dyn_cast<triton::nvidia_gpu::CLCTryCancelOp>(op)) {
49+
return clcTryCancelOp.getResult();
50+
}
4751
return Value();
4852
}
4953

0 commit comments

Comments
 (0)