[Mlir-commits] [mlir] [mlir][linalg] Add support for scalable vectorization of linalg.mmt4d (PR #146531)

Diego Caballero llvmlistbot at llvm.org
Tue Jul 8 09:14:38 PDT 2025


================
@@ -871,7 +871,8 @@ FailureOr<VectorizationResult>
 vectorize(RewriterBase &rewriter, Operation *op,
           ArrayRef<int64_t> inputVectorSizes = {},
           ArrayRef<bool> inputScalableVecDims = {},
-          bool vectorizeNDExtract = false, bool flatten1DDepthwiseConv = false);
+          bool vectorizeNDExtract = false, bool flatten1DDepthwiseConv = false,
+          bool assumeScalableSizesMultipleOfDim = false);
----------------
dcaballe wrote:

We have `assumeScalableSizesMultipleOfDim` and `getAssumeScalableSizesMatchDimSize`. Should we have only one?
Also, it should be "dim multiple of vector sizes"?

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


More information about the Mlir-commits mailing list