[Mlir-commits] [mlir] [mlir][linalg] Fix empty outer dim case for packing reshape op (PR #96732)
Adam Siemieniuk
llvmlistbot at llvm.org
Wed Jun 26 01:03:21 PDT 2024
================
@@ -641,7 +641,14 @@ bubbleUpPackOpThroughCollapseShape(tensor::CollapseShapeOp collapseOp,
PatternRewriter &rewriter) {
SmallVector<int64_t> innerTileSizes = packOp.getStaticTiles();
ArrayRef<int64_t> innerDimsPos = packOp.getInnerDimsPos();
- ArrayRef<int64_t> outerDimsPerm = packOp.getOuterDimsPerm();
+ auto numOuterDims =
+ dyn_cast<RankedTensorType>(packOp.getDpsInputs()[0].getType())
----------------
adam-smnk wrote:
nit: `packOp.getSourceType()`
https://github.com/llvm/llvm-project/pull/96732
More information about the Mlir-commits
mailing list