[Mlir-commits] [llvm] [mlir] [MLIR] Generalize expand_shape to take shape as explicit input (PR #69267)
llvmlistbot at llvm.org
llvmlistbot at llvm.org
Wed Apr 17 16:36:55 PDT 2024
================
@@ -757,7 +757,10 @@ pushDownUnPackOpThroughExpandShape(tensor::UnPackOp unPackOp,
ArrayRef<int64_t> innerDimsPos = unPackOp.getInnerDimsPos();
ArrayRef<int64_t> outerDimsPerm = unPackOp.getOuterDimsPerm();
- ArrayRef<int64_t> dstShape = expandOp.getType().getShape();
+ auto expandTy = expandOp.getType().dyn_cast<RankedTensorType>();
----------------
MaheshRavishankar wrote:
Why did you have to change this? This should have always been a ranked tensor type. If so then something is off with the pack op. Fine to leave as is, just noticing.
https://github.com/llvm/llvm-project/pull/69267
More information about the Mlir-commits
mailing list