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

[Matthias Springer]

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

This pattern has an additional `isShapePreserving` check that makes it safe. Btw, that check is quite conservative, it could be improved by checking for `DestinationStyleOpInterface` instead of hard-coding a few ops in the analysis.

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