Skip to content

Commit aa35313

Browse files
committed
fixup! [mlir][linalg] Fix vectorization of tensor.extract
Fix SmallVector initialization
1 parent 47376c5 commit aa35313

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

mlir/lib/Dialect/Linalg/Transforms/Vectorization.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1168,7 +1168,7 @@ vectorizeTensorExtract(RewriterBase &rewriter, VectorizationState &state,
11681168
// Mask this broadcasting xfer_read here rather than relying on the generic
11691169
// path (the generic path assumes identity masking map, which wouldn't be
11701170
// valid here).
1171-
SmallVector<int64_t> readMaskShape{1};
1171+
SmallVector<int64_t> readMaskShape = {1};
11721172
auto readMaskType = VectorType::get(readMaskShape, rewriter.getI1Type());
11731173
auto allTrue = rewriter.create<vector::ConstantMaskOp>(
11741174
loc, readMaskType, vector::ConstantMaskKind::AllTrue);

0 commit comments

Comments
 (0)