[Mlir-commits] [mlir] [mlir][tosa] Change the start and size of slice to tosa shape type (PR #124209)
Georgios Pinitas
llvmlistbot at llvm.org
Fri Jan 24 09:52:21 PST 2025
================
@@ -282,17 +298,27 @@ class SliceConverter : public OpConversionPattern<tosa::SliceOp> {
auto dim = rewriter.create<tensor::DimOp>(loc, input, index);
auto offset = rewriter.create<arith::ConstantOp>(
- loc, rewriter.getIndexAttr(starts[index]));
+ loc, rewriter.getIndexAttr(SliceStarts[index]));
dynSizes.push_back(rewriter.create<arith::SubIOp>(loc, dim, offset));
}
auto newSliceOp = rewriter.create<tensor::ExtractSliceOp>(
sliceOp.getLoc(), sliceOp.getType(), input, ValueRange({}), dynSizes,
- ValueRange({}), rewriter.getDenseI64ArrayAttr(starts),
+ ValueRange({}), rewriter.getDenseI64ArrayAttr(SliceStarts),
rewriter.getDenseI64ArrayAttr(sizes),
rewriter.getDenseI64ArrayAttr(strides));
rewriter.replaceOp(sliceOp, newSliceOp.getResult());
+
+ auto removeIfRedundant = [&](Operation *op) {
----------------
GeorgeARM wrote:
Do you plan to use in other operations? If yes probably worth extracting to a common function?
https://github.com/llvm/llvm-project/pull/124209
More information about the Mlir-commits
mailing list