reduced some spilling

2023-01-02 19:28:54 -08:00
parent c11fe351e1
commit 05920e0b8b
3 changed files with 152 additions and 148 deletions
--- a/lib/Dialect/TritonGPU/Transforms/Combine.cpp
+++ b/lib/Dialect/TritonGPU/Transforms/Combine.cpp
@@ -483,7 +483,7 @@ public:
      return op->getBlock() == cvt->getBlock() &&
             !(isa<triton::ReduceOp>(op) &&
               !op->getResult(0).getType().isa<RankedTensorType>()) &&
-             !isa<triton::gpu::ConvertLayoutOp>(op) &&
+            //  !isa<triton::gpu::ConvertLayoutOp>(op) &&
             !isa<scf::YieldOp>(op);
    };
    mlir::getForwardSlice(cvt.getResult(), &cvtSlices, filter);