summaryrefslogtreecommitdiff
path: root/mlir/lib/Dialect/Vector/Transforms/VectorInsertExtractStridedSliceRewritePatterns.cpp
diff options
context:
space:
mode:
authorAndrzej WarzyƄski <andrzej.warzynski@arm.com>2025-09-15 21:18:26 +0100
committerGitHub <noreply@github.com>2025-09-15 21:18:26 +0100
commit1287ed1fa2dbf81ce2549a4cb5ae339857c778f9 (patch)
treeec1d4bb2c5d008218e561eba09faff29b5bf5082 /mlir/lib/Dialect/Vector/Transforms/VectorInsertExtractStridedSliceRewritePatterns.cpp
parent04cd39ae287d2c35d2b64cb70ea7bcba7e9796d9 (diff)
[mlir][vector] Use `source` as the source argument name (#158258)
This patch updates the following ops to use `source` (instead of `vector`) as the name for their source argument: * `vector.extract` * `vector.scalable.extract` * `vector.extract_strided_slice` This change ensures naming consistency with the "builders" for these Ops that already use the name `source` rather than `vector`. It also addresses part of: * https://github.com/llvm/llvm-project/issues/131602 Specifically, it ensures that we use `source` and `dest` for read and write operations, respectively (as opposed to `vector` and `dest`).
Diffstat (limited to 'mlir/lib/Dialect/Vector/Transforms/VectorInsertExtractStridedSliceRewritePatterns.cpp')
-rw-r--r--mlir/lib/Dialect/Vector/Transforms/VectorInsertExtractStridedSliceRewritePatterns.cpp8
1 files changed, 4 insertions, 4 deletions
diff --git a/mlir/lib/Dialect/Vector/Transforms/VectorInsertExtractStridedSliceRewritePatterns.cpp b/mlir/lib/Dialect/Vector/Transforms/VectorInsertExtractStridedSliceRewritePatterns.cpp
index cbb9d4bbf0b1..f6d6555f4c6e 100644
--- a/mlir/lib/Dialect/Vector/Transforms/VectorInsertExtractStridedSliceRewritePatterns.cpp
+++ b/mlir/lib/Dialect/Vector/Transforms/VectorInsertExtractStridedSliceRewritePatterns.cpp
@@ -213,8 +213,8 @@ public:
for (int64_t off = offset, e = offset + size * stride; off < e;
off += stride)
offsets.push_back(off);
- rewriter.replaceOpWithNewOp<ShuffleOp>(op, dstType, op.getVector(),
- op.getVector(), offsets);
+ rewriter.replaceOpWithNewOp<ShuffleOp>(op, dstType, op.getSource(),
+ op.getSource(), offsets);
return success();
}
};
@@ -250,7 +250,7 @@ public:
SmallVector<Value> elements;
elements.reserve(size);
for (int64_t i = offset, e = offset + size * stride; i < e; i += stride)
- elements.push_back(ExtractOp::create(rewriter, loc, op.getVector(), i));
+ elements.push_back(ExtractOp::create(rewriter, loc, op.getSource(), i));
Value result = arith::ConstantOp::create(
rewriter, loc, rewriter.getZeroAttr(op.getType()));
@@ -306,7 +306,7 @@ public:
Value res = BroadcastOp::create(rewriter, loc, dstType, zero);
for (int64_t off = offset, e = offset + size * stride, idx = 0; off < e;
off += stride, ++idx) {
- Value one = ExtractOp::create(rewriter, loc, op.getVector(), off);
+ Value one = ExtractOp::create(rewriter, loc, op.getSource(), off);
Value extracted = ExtractStridedSliceOp::create(
rewriter, loc, one, getI64SubArray(op.getOffsets(), /* dropFront=*/1),
getI64SubArray(op.getSizes(), /* dropFront=*/1),