[Mlir-commits] [mlir] [mlir][transform] Fix failure in flattening already flattened linalg ops (PR #86037)
llvmlistbot at llvm.org
llvmlistbot at llvm.org
Thu Mar 21 09:39:54 PDT 2024
================
@@ -3269,15 +3269,20 @@ DiagnosedSilenceableFailure transform::FlattenElementwiseLinalgOp::applyToOne(
transform::ApplyToEachResultList &results,
transform::TransformState &state) {
rewriter.setInsertionPoint(target);
- if (target.getNumLoops() <= 1)
+ if (!isElementwise(target)) {
+ failed(rewriter.notifyMatchFailure(
+ target, "only elementwise flattening is supported"));
+ return emitDefaultSilenceableFailure(target);
----------------
srcarroll wrote:
appreciate it. thanks for teaching me. :)
https://github.com/llvm/llvm-project/pull/86037
More information about the Mlir-commits
mailing list