[Mlir-commits] [mlir] [mlir][tosa] Add support for 0 dim in ReshapeOp (PR #84004)

Georgios Pinitas llvmlistbot at llvm.org
Wed Jan 22 05:27:08 PST 2025


================
@@ -928,10 +928,16 @@ LogicalResult tosa::ReshapeOp::inferReturnTypeComponents(
   // dimension.
   int64_t numElements = inputShape.getNumElements();
   int64_t staticMul = 1;
-  for (auto val : newShapeValue) {
+  int i = 0;
+  for (auto &val : newShapeValue) {
+    if (val == 0) {
----------------
GeorgeARM wrote:

Can a shape value be 0?

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


More information about the Mlir-commits mailing list