[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