[Mlir-commits] [mlir] [mlir] Fix bug in UnPackOp tiling implementation causing infinite loop (PR #113571)
llvmlistbot at llvm.org
llvmlistbot at llvm.org
Thu Oct 24 09:31:49 PDT 2024
================
@@ -554,13 +554,14 @@ struct UnPackOpTiling
sliceSrcIndices.append(numInnerTiles, zeroAttr);
sliceSrcSizes.append(unpackOp.getMixedTiles());
sliceSrcStrides.append(numInnerTiles, oneAttr);
- Value sliceSource =
+ SmallVector<Operation *> generatedSlices;
+ ExtractSliceOp sliceSource =
b.create<ExtractSliceOp>(loc, unpackOp.getSource(), sliceSrcIndices,
sliceSrcSizes, sliceSrcStrides);
+ generatedSlices.push_back(sliceSource);
----------------
Max191 wrote:
I'm realizing that maybe I'm not supposed to add the input operand slices. Should I be doing this?
https://github.com/llvm/llvm-project/pull/113571
More information about the Mlir-commits
mailing list