[Mlir-commits] [mlir] [mlir][vector] drop unit dim from memrefs for xfer_read/write with non-reduced mask (PR #187076)

[Ege Beysel]

================
@@ -532,12 +532,15 @@ class TransferReadDropUnitDimsPattern
       }
       FailureOr<Value> rankReducedCreateMask =
           createMaskDropNonScalableUnitDims(rewriter, loc, createMaskOp);
-      if (failed(rankReducedCreateMask)) {
+      if (succeeded(rankReducedCreateMask)) {
+        maskOp = *rankReducedCreateMask;
+        LDBG() << "  -> Successfully reduced mask dimensions";
+      } else if (createMaskOp.getVectorType().getRank() !=
+                 reducedVectorType.getRank()) {
+        // Mask needs reduction but couldn't be reduced.
         LDBG() << "  -> Failed to reduce mask dimensions";
----------------
egebeysel wrote:

done

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