[Mlir-commits] [mlir] [MLIR][Linalg] Scalable Vectorization of Reduction (PR #97788)
Andrzej WarzyĆski
llvmlistbot at llvm.org
Thu Jul 11 07:24:23 PDT 2024
================
@@ -1947,13 +1956,30 @@ vectorizeScalableVectorPrecondition(Operation *op,
if (inputVectorSizes.empty())
return success();
+ auto linalgOp = dyn_cast<LinalgOp>(op);
+ if (linalgOp && isLinalgReduction(linalgOp)) {
+ LDBG("Checking reduce op dims for scalable vectorization\n");
+ auto iteratorTypes = linalgOp.getIteratorTypesArray();
+ assert(iteratorTypes.size() == inputScalableVecDims.size() &&
+ "Number of iterator types and input scalable dims mismatch");
----------------
banach-space wrote:
Isn't this already "effectively" covered by the assert on L1944?
https://github.com/llvm/llvm-project/pull/97788
More information about the Mlir-commits
mailing list