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

    <tr>
        <th>Summary</th>
        <td>
            [MLIR] Inconsistent output when executing MLIR program with and without `-test-loop-fusion=test-loop-fusion-transformation`
        </td>
    </tr>

    <tr>
      <th>Labels</th>
      <td>
            mlir
      </td>
    </tr>

    <tr>
      <th>Assignees</th>
      <td>
      </td>
    </tr>

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

<pre>
    My git version is [ea44647](https://github.com/llvm/llvm-project/commit/ea44647a0b49de826191eeb6e05020262b5a81e9).

## Description:
I am experiencing an inconsistent result when executing the same MLIR program with and without the `-test-loop-fusion=test-loop-fusion-transformation`.

## Steps to Reproduce:

### 1. **MLIR Program (test.mlir)**:

test.mlir:

```
module {
  func.func private @printMemrefI64(tensor<*xi64>)
  func.func @main() -> () {
    %arg0 = "tosa.const"() {value = dense<20> : tensor<36x2x3xi64>} : () -> tensor<36x2x3xi64>
    %0 = tosa.reduce_sum %arg0 {axis = 0 : i32} : (tensor<36x2x3xi64>) -> tensor<1x2x3xi64>
    %rtn1 = tensor.cast %0 : tensor<1x2x3xi64> to tensor<*xi64>
    call @printMemrefI64(%rtn1) : (tensor<*xi64>) -> ()
    return
  }
}
```

### 2. **Command to Run Without `-test-loop-fusion=test-loop-fusion-transformation`:**

```
/path/llvm-project/build/bin/mlir-opt test.mlir -pass-pipeline='builtin.module(func.func(tosa-to-linalg-named,tosa-to-linalg))' | \
/path/llvm-project/build/bin/mlir-opt -tosa-to-arith -one-shot-bufferize="bufferize-function-boundaries" -convert-linalg-to-affine-loops -lower-affine -convert-scf-to-cf -expand-strided-metadata -convert-cf-to-llvm -convert-arith-to-llvm -finalize-memref-to-llvm -convert-func-to-llvm -reconcile-unrealized-casts | \
/path/llvm-project/build/bin/mlir-cpu-runner -e main -entry-point-result=void \
-shared-libs=/path/llvm-project/build/lib/libmlir_runner_utils.so \
-shared-libs=/path/llvm-project/build/lib/libmlir_c_runner_utils.so \
-shared-libs=/path/llvm-project/build/lib/libmlir_async_runtime.so
```

### 3. **Output Without `-test-loop-fusion=test-loop-fusion-transformation`:**

```
[[[720, 720,    720], 
  [720,    720,    720]]]
```

### 4. **Command to Run With `-test-loop-fusion=test-loop-fusion-transformation`:**

```
/path/llvm-project/build/bin/mlir-opt test.mlir -pass-pipeline='builtin.module(func.func(tosa-to-linalg-named,tosa-to-linalg))' | \
/path/llvm-project/build/bin/mlir-opt -tosa-to-arith -one-shot-bufferize="bufferize-function-boundaries" -convert-linalg-to-affine-loops -test-loop-fusion=test-loop-fusion-transformation -lower-affine -convert-scf-to-cf -expand-strided-metadata -convert-cf-to-llvm -convert-arith-to-llvm -finalize-memref-to-llvm -convert-func-to-llvm -reconcile-unrealized-casts | \
/path/llvm-project/build/bin/mlir-cpu-runner -e main -entry-point-result=void \
-shared-libs=/path/llvm-project/build/lib/libmlir_runner_utils.so \
-shared-libs=/path/llvm-project/build/lib/libmlir_c_runner_utils.so \
-shared-libs=/path/llvm-project/build/lib/libmlir_async_runtime.so
```

### 5. **Output With `-test-loop-fusion=test-loop-fusion-transformation`:**

```
[[[20,    20, 20], 
  [20,    20,    20]]]
```

I'm not sure if there is any bug in my program or if the wrong usage of the above passes caused this result.
</pre>
<img width="1" height="1" alt="" src="http://email.email.llvm.org/o/eJzsV0uPozgQ_jXOpWQE5hUOOaSTidTSjHY1e9jjyEBBvAIb-ZFO769f2ZCkX9Orae1IexjJAsd8rvooV32huDGil4gbkt-RfL_izh6V3nzmY600y1a1ah83Xx6hFxZOqI1QEoQBkt8hz7IiK0m-J2x9tHYyJN0SdiDs0At7dHXUqJGwwzCcLjc6afUXNpawQ6PGUfjJYobHdVa1uGZFUiWIdYFxHrOYFazO-TrBirAqIvHWD5YSlsIeTaPFZIWS3nG8vQc-Ap4n1AJlI2QPXIKQjZJGGIvSgkbjBgsPR5SAZ2yc9Sh7RDB8RPjy-f4rTFr1mo_wIOwRuGzDRDkbYKSIqUVj6aDURDtngvP9yyVqNZemU3rkgV4RP6f-h8XJgFXwFSetWtfg_AZXhAclERC2JWwbaP2-0CJs7b1F4yA0YdWMuO6-PbraK-JlxNtRtW5AIOUdibcAnZNN5C8waXHiFoFk8aSFtF9w1NjdF1nwJo3SJN0Rtj2LIiPpJ-_3uQGSxSMXkrA1YRVQkn6CZX5xBkBYznUfA0n3QBizyvDIn40ljN3AJz44DJgWpUGS7lgczKVbuFJJizM7pxc25T48fer8O8grkZlFoKDRx_-bceONYXnHzz7J0z3EwbZI2RM337H-ynnypm9tZTK7D7io4cZeOG3f3uxT5a1zmG02fBjePLrFWYjsC-bPDvPpgS02NVqnZfhByr3Po_n6JJueJSu7JOtOjaOvGp_cTsKfS_F8uHCCpoTxKp0JO0zcHl9rS-3E0Pq7z8iDLweqJgvX4gA6cWPoJCYchESS7gkr_SYrZDQXCWHra3b7uCnDqVV0EJIPPZV8xJaw3fPlUI4VYSWQcgck332AI72Y5NrrD1USqTkqS2vXdajF3zNbdv1JPUMfK1orJ1uuBRrCGNBGyRNqe6HsTXadkBjCbYAO6gH1snZDm6bz0KYDiueJy5Yaq0WLLR3R8pZbfsPOUP9et7VA-7bceeee5RjS8jXes7-tamyUbMSA1EmNYWdLfX2YD4a0mRzVTkrUQBG8RAFFafUjnZSQls5_CCTdn5RoF_vUHLnGlg6iNiHY_-JuEPV89R6_zd6-OSsGExn1n9hsfopVbh5lsGzFiJFR75V3einv35ydnP25Ze0_RPwoWUzYDpYbQJj5r40dzMJ0hSzPnqHm8c47Ze9I1i-9-t_p1Y-fxi-N-6VxP6Bx-Rsa93MF7ipZ8-SVur14Pk--L233hJUjSGXBOI0gOt-y-IkBLh-hdj0ICePjtcVRegHBg1ayB2d4j6DmJV6rE4IXHjTQcGewBXsUZmmiolW7SdsqrfgKN0mZZnFV5lW1Om6wwaZJknVW1lVSZVh1PK_KNmVVUlVZka7EhsUsS1iSJmlWpFUUV5yvMYnzuMuaYl2RLMaRiyHyZxwp3a-EMQ43SVKty2I18BoHExpWxuY-iPnWVW9CTtSuNySLB2GsuVmwwg6hyfXdFMn3cP-0MVTzkb9oDN_vBz-cGiunh80PN8whBIawwxKF04b9EwAA__-Zr_t8">