[Mlir-commits] [mlir] [mlir][vector] Add support for scalable vectors to VectorLinearize (PR #86786)
Andrzej Warzyński
llvmlistbot at llvm.org
Thu Mar 28 04:23:54 PDT 2024
================
@@ -317,3 +317,8 @@ SmallVector<OpFoldResult> vector::getMixedSizesXfer(bool hasTensorSemantics,
: memref::getMixedSizes(rewriter, loc, base);
return mixedSourceDims;
}
+
+bool vector::isLinearizableVector(VectorType type) {
+ auto numScalableDims = llvm::count(type.getScalableDims(), true);
+ return ((type.getRank() > 1) && (numScalableDims <= 1));
----------------
banach-space wrote:
I'd like to keep the inner parentheses - I find that easier to parse. Probably a subjective thing 😅
https://github.com/llvm/llvm-project/pull/86786
More information about the Mlir-commits
mailing list