[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