.

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