[OPTIMIZER] Fix crash in loop pipelining. (triton-lang#1602)

Fixes issue triton-lang#1601.
siliconflow · May 2, 2023 · ea02bfc · ea02bfc
1 parent 70cdf21
commit ea02bfc
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/lib/Dialect/TritonGPU/Transforms/Pipeline.cpp b/lib/Dialect/TritonGPU/Transforms/Pipeline.cpp
@@ -579,7 +579,12 @@ scf::ForOp LoopPipeliner::createNewForOp() {
     // we replace the use new load use with a convert layout
     size_t i = std::distance(loads.begin(), it);
     auto cvtDstTy = op.getResult(0).getType().cast<RankedTensorType>();
-    auto cvtDstEnc = cvtDstTy.getEncoding().cast<ttg::DotOperandEncodingAttr>();
+    auto cvtDstEnc =
+        cvtDstTy.getEncoding().dyn_cast<ttg::DotOperandEncodingAttr>();
+    if (!cvtDstEnc) {
+      builder.clone(op, mapping);
+      continue;
+    }
     auto newDstTy = RankedTensorType::get(
         cvtDstTy.getShape(), cvtDstTy.getElementType(),
         ttg::DotOperandEncodingAttr::get(