[Mlir-commits] [mlir] [mlir][tosa] Fix ability to expand ranks with dynamic shape support (PR #128037)

llvmlistbot at llvm.org llvmlistbot at llvm.org
Mon Feb 24 10:52:12 PST 2025


================
@@ -88,9 +88,9 @@ computeReshapeOutput(ArrayRef<int64_t> higherRankShape,
     higherRankDim = higherRankShape[i];
     lowerRankDim = lowerRankShape[j];
 
-    if (lowerRankDim == 1 && higherRankDim > 1)
+    if (lowerRankDim == 1 && higherRankDim != 1)
----------------
Jerry-Ge wrote:

what do you specifically mean right aligning? I think we still need some logic to implement different special cases instead of adding 1s to the left of lower rank tensor shape? 
For example: 

low Rank: `[b, c]`
high Rank: `[a, 1, c]`
the output shape would be `[a, b, c]` given the existing logic.

we can't simply adding 1s to the left of `[b, c]` (right aligning?) it would be `[1, b, c]` which is not what we wanted. 



https://github.com/llvm/llvm-project/pull/128037


More information about the Mlir-commits mailing list