[Mlir-commits] [mlir] [mlir][transform] Fix failure in flattening already flattened linalg ops (PR #86037)

Oleksandr Alex Zinenko llvmlistbot at llvm.org
Thu Mar 21 09:34:34 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);
----------------
ftynse wrote:

No big deal, the point of review is also for contributors to learn things.

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


More information about the Mlir-commits mailing list