[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