[OPTIMIZER] Added DotOp to the list of expensive ops we don't want to rematerialize. (#718)

2022-09-27 09:05:49 -07:00
parent df8d276089
commit 9ddf0921fb
1 changed files with 2 additions and 2 deletions
--- 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<triton::gpu::ExtractSliceOp, triton::gpu::AllocTensorOp,
-              triton::gpu::InsertSliceAsyncOp, triton::LoadOp, triton::StoreOp>(
-              op))
+              triton::gpu::InsertSliceAsyncOp, triton::LoadOp, triton::StoreOp,
+              triton::DotOp>(op))
        return true;
      if (isa<scf::YieldOp, scf::ForOp>(op))
        return true;