diff --git a/lib/Dialect/TritonGPU/Transforms/Combine.cpp b/lib/Dialect/TritonGPU/Transforms/Combine.cpp index d879f451e..a354993a8 100644 --- a/lib/Dialect/TritonGPU/Transforms/Combine.cpp +++ b/lib/Dialect/TritonGPU/Transforms/Combine.cpp @@ -120,8 +120,8 @@ public: auto blacklist = [](Operation *op) { if (isa( - op)) + triton::gpu::InsertSliceAsyncOp, triton::LoadOp, triton::StoreOp, + triton::DotOp>(op)) return true; if (isa(op)) return true;