<table border="1" cellspacing="0" cellpadding="8">
    <tr>
        <th>Issue</th>
        <td>
            <a href=https://github.com/llvm/llvm-project/issues/87396>87396</a>
        </td>
    </tr>

    <tr>
        <th>Summary</th>
        <td>
            TosaToTensor regression from #85798
        </td>
    </tr>

    <tr>
      <th>Labels</th>
      <td>
            regression
      </td>
    </tr>

    <tr>
      <th>Assignees</th>
      <td>
            sabauma,
            sjarus,
            rafaelubalmw
      </td>
    </tr>

    <tr>
      <th>Reporter</th>
      <td>
          bjacob
      </td>
    </tr>
</table>

<pre>
    The following test starts failing as of #85798, with a failing assertion introduced in that PR, and passes when it is reverted:

```mlir
 func.func @mlp_invocation(%lhs: tensor<2x4xf32>, %rhs : tensor<4x8xf32>) -> tensor<2x8xf32> {
    %lhs_3D = tosa.reshape %lhs {new_shape = array<i64 : 1, 2, 2>} : (tensor<2x4xf32>) -> tensor<1x2x4xf32>
    %rhs_3D = tosa.reshape %rhs {new_shape = array<i64 : 1, 2, 2>} : (tensor<4x8xf32>) -> tensor<1x4x8xf32>
    %0 = tosa.matmul %lhs_3D, %rhs_3D : (tensor<1x2x4xf32>, tensor<1x4x8xf32>) -> tensor<1x2x8xf32>
    %1 = tosa.clamp %0 {
        min_int = 0 : i64, max_int = 9223372036854775807 : i64,
        min_fp = 0.0 : f32, max_fp = 3.4028235e+38 : f32}
 : (tensor<1x2x8xf32>) -> tensor<1x2x8xf32>
    %2 = tosa.negate %1 : (tensor<1x2x8xf32>) -> tensor<1x2x8xf32>
    %3 = tosa.reshape %2 {new_shape = array<i64 : 2, 2>}  : (tensor<1x2x8xf32>) -> tensor<2x8xf32>
 return %3 : tensor<2x8xf32>
 }
```

Run:

```
mlir-opt --tosa-to-tensor /tmp/a.mlir
```

Result:

```
mlir-opt: /home/benoit/megabump/work/iree/third_party/llvm-project/mlir/lib/Conversion/TosaToTensor/TosaToTensor.cpp:196: SmallVector<ReassociationExprs> (anonymous namespace)::createReassociationMapForCollapse(OpBuilder &, Type, Type): Assertion `currSrcDim == srcShape.size() && currDstDim == dstShape.size()' failed.
```

</pre>
<img width="1px" height="1px" alt="" src="http://email.email.llvm.org/o/eJysVs1u6zgPfRplI8SwKcdWFlmkSb378A3aYrYFbTOxOrJlSHKTztMPZKf5uU3v_OAGgQKJR-TRIUEGnVP7jmjFFg8MwGGJQ4sMgMEm7N_QDu68tbhD0kOJuj2Ew8V2hoNvjF2Vb1iZclaa-mP10hDfGa3NQXV77sl57jxa7_gOlQ5n6LjZcQZCLvKlZLDhB-UbjlcAR9Yr03HVeWvqoaKaq477Bj3_7SncwK7mfcA5fmio48pz5bild7KeaibWLN6y-HPN4unbamWnI74buioKC2dp3Or-VXXvpsIQlYFksNCNY2LNPXXOWCY2cEyPOwFMPIb4DBa2cfwGkR7lGbHkcyYer29_2jjLH04cOOdToFex5UxsuTcOI0uuwZ5OpgDv6PB6OhNbjtbiBxMblaVj_CTwgWkRjyzfjqcM5F3qPxJLjlfWa1r2e1r219D6mV7J8cp6TSu-MGrRt4O-SHjJy0T8NtrNQ2HzTai7-twlklyIVBrb_kTuOrnh06ruVXV-BMcjKZWlIX6Lx7NhCSBEDrHI5CLN84WM8yvoV4e7fvIXTR4Du5PHk0VEaQwSxIIYPAh5RuXbk7N76vwHEeAiQkd79PSpzC_xLu5WH_x97d2U3b9k84WLJT_Y7pPP-ifIs7znhnPdhZ6G7ru-NG1Dd5qb3vP5PLx57s18isUZFL7tGRQYXVrY_SDkBu3_WZxJl6IxLTEoSuqM8gyKlvZYDmO4g7F_MCiUpYDwjbL1a4_WfzAotH5v5701b1SNtwIvKLQqGRQb072TdWMvLV6MwxfzMql2u42qvmdinSyzwOW5Ra1_p8qP6j4ROmcqNbbkx2Nv3dg8QWJnuo_WDI532JLrsSIGy_Bksa4soaebq__DvjB2Y7TG3hED-f_-YVC6piBrFgrl5aOny2_wxNfnCcSyuBqsfbbVVrWh2EK9OVs9h-KLnPqTxnGxnJxlPIC3zl-Ba-d_BDPIx2FHdXQ3Q7N6JeqlWOKMVkmeQCzTRIpZs5J1XSEusc5rQClhEWdlklCFJMtKgJipFcSQxmkMiVxkIKKljBEywCopJZJElsbUotJRyF9k7H6mnBtoJXOxzGYaS9Lu9HfA0t6Sm7I4Tnu7GpNeDnvH0lgr593FjVde0-o6u_zigO-sac8jfzZYvWq878OEZVAwKPbKN0MZVaY9ldbXCht5OgbFSPWvAAAA__8i9IdA">