[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