Skip to content

Commit 859ce66

Browse files
committed
Address comments
1 parent 7196ea3 commit 859ce66

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

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

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1414,7 +1414,7 @@ vectorizeAsLinalgGeneric(RewriterBase &rewriter, VectorizationState &state,
14141414
// TODO: remove this.
14151415
if (readType.getRank() == 0)
14161416
readValue = rewriter.create<vector::ExtractOp>(loc, readValue,
1417-
SmallVector<int64_t>{});
1417+
ArrayRef<int64_t>());
14181418

14191419
LDBG("New vectorized bbarg(" << bbarg.getArgNumber() << "): " << readValue
14201420
<< "\n");
@@ -2272,8 +2272,8 @@ LogicalResult mlir::linalg::vectorizeCopy(RewriterBase &rewriter,
22722272
loc, readType, copyOp.getSource(), indices,
22732273
rewriter.getMultiDimIdentityMap(srcType.getRank()));
22742274
if (cast<VectorType>(readValue.getType()).getRank() == 0) {
2275-
readValue = rewriter.create<vector::ExtractOp>(loc, readValue,
2276-
SmallVector<int64_t>{});
2275+
readValue =
2276+
rewriter.create<vector::ExtractOp>(loc, readValue, ArrayRef<int64_t>());
22772277
readValue = rewriter.create<vector::BroadcastOp>(loc, writeType, readValue);
22782278
}
22792279
Operation *writeValue = rewriter.create<vector::TransferWriteOp>(

0 commit comments

Comments
 (0)