[Mlir-commits] [mlir] [MLIR] Enable pattern only for scf.forall op (PR #110230)

Matthias Springer llvmlistbot at llvm.org
Fri Sep 27 11:49:45 PDT 2024


================
@@ -131,11 +132,25 @@ struct IterArgsToInitArgs : public OpRewritePattern<tensor::DimOp> {
     auto blockArg = dyn_cast<BlockArgument>(dimOp.getSource());
     if (!blockArg)
       return failure();
-    auto loopLikeOp =
-        dyn_cast<LoopLikeOpInterface>(blockArg.getParentBlock()->getParentOp());
-    if (!loopLikeOp)
+    // TODO: Enable this for loopLikeInterface. Restricting for scf.for
----------------
matthias-springer wrote:

fyi, for `scf.for` we already have this pattern here: https://github.com/llvm/llvm-project/blob/main/mlir/lib/Dialect/SCF/Transforms/LoopCanonicalization.cpp#L87

https://github.com/llvm/llvm-project/pull/110230


More information about the Mlir-commits mailing list