[Mlir-commits] [mlir] [mlir][vector] Fix unit dim dropping pattern for masked writes (PR #74038)

Diego Caballero llvmlistbot at llvm.org
Fri Dec 1 05:19:20 PST 2023


================
@@ -461,25 +451,39 @@ class TransferWriteDropUnitDimsPattern
       return failure();
     // Check if the reduced vector shape matches the reduced destination shape.
     // Otherwise, this case is not supported yet.
-    int vectorReducedRank = getReducedRank(vectorType.getShape());
-    if (reducedRank != vectorReducedRank)
+    auto reducedVectorType = trimNonScalableUnitDims(vectorType);
----------------
dcaballe wrote:

spell auto

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


More information about the Mlir-commits mailing list