================
@@ -434,8 +429,21 @@ FailureOr<LowerUnPackOpResult>
linalg::lowerUnPack(RewriterBase &rewriter,
RankedTensorType::Builder(packedTensorType).setShape(stripMinedShape);
RankedTensorType collapsedType = tensor::CollapseShapeOp::inferCollapsedType(
stripMinedTensorType, packingMetadata.reassociations);
+
+ // Get dynamic dims from input tensor in order of stripMinedTensor
+ // `tensor.empty` op
+ SmallVector<OpFoldResult, 4> dims =
+ tensor::getMixedSizes(rewriter, loc, unPackOp.getSource());
+ applyPermutationToVector(dims, lastDimsToInsertPositionsPerm);
+ SmallVector<Value, 4> dynDims;
----------------
srcarroll wrote:
oh nice. i wasn't aware of this builder. very convenient.
https://github.com/llvm/llvm-project/pull/75494
_______________________________________________
cfe-commits mailing list
cfe-commits@lists.llvm.org
https://lists.llvm.org/cgi-bin/mailman/listinfo/cfe-commits
