[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