diff options
| author | Andrzej WarzyĆski <andrzej.warzynski@arm.com> | 2025-09-15 21:18:26 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2025-09-15 21:18:26 +0100 |
| commit | 1287ed1fa2dbf81ce2549a4cb5ae339857c778f9 (patch) | |
| tree | ec1d4bb2c5d008218e561eba09faff29b5bf5082 /mlir/lib/Dialect/Vector/Transforms/VectorInsertExtractStridedSliceRewritePatterns.cpp | |
| parent | 04cd39ae287d2c35d2b64cb70ea7bcba7e9796d9 (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.cpp | 8 |
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), |
