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

[Prashant Kumar]

================
@@ -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
----------------
pashu123 wrote:

I'll be merging this and refactoring in upcoming PRs.

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