[Mlir-commits] [mlir] [mlir][Vectorizer] Added support to Vectorize tensor.unpack (PR #76087)

llvmlistbot at llvm.org llvmlistbot at llvm.org
Wed Feb 14 13:22:15 PST 2024


================
@@ -1420,16 +1420,28 @@ static Value createReadOrMaskedRead(OpBuilder &builder, Location loc,
   auto sourceShape = dyn_cast<ShapedType>(source.getType()).getShape();
   assert(sourceShape.size() == readShape.size());
   auto maskType = VectorType::get(readShape, builder.getI1Type());
-  auto vectorType = VectorType::get(readShape, padValue.getType());
+  Type vecElemType = padValue != nullptr
----------------
Max191 wrote:

I think we can create a new constant value for unpack, similar to the case when there is no padding_value for pack:
https://github.com/llvm/llvm-project/blob/1301bc46aea14297478bd13bcacff429e2a18c04/mlir/lib/Dialect/Linalg/Transforms/Vectorization.cpp#L1516-L1519

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


More information about the Mlir-commits mailing list