[Mlir-commits] [mlir] [mlir][tensor] Fix off-by-one error in ReshapeOpsUtils (PR #112774)

Prashant Kumar llvmlistbot at llvm.org
Thu Oct 17 22:41:26 PDT 2024


================
@@ -47,7 +47,7 @@ mlir::getReassociationIndicesForCollapse(ArrayRef<int64_t> sourceShape,
       break;
 
     int64_t currTargetShape = targetShape[targetDim];
-    while (sourceDim < sourceShape.size() &&
+    while (sourceDim < sourceShape.size() - 1 &&
----------------
pashu123 wrote:

```suggestion
    while (sourceDim < (sourceShape.size() - 1) &&
```

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


More information about the Mlir-commits mailing list