<table border="1" cellspacing="0" cellpadding="8">
<tr>
<th>Issue</th>
<td>
<a href=https://github.com/llvm/llvm-project/issues/131105>131105</a>
</td>
</tr>
<tr>
<th>Summary</th>
<td>
[mlir] Inconsistent output when executing MLIR program with `--test-affine-parametric-tile`
</td>
</tr>
<tr>
<th>Labels</th>
<td>
mlir
</td>
</tr>
<tr>
<th>Assignees</th>
<td>
</td>
</tr>
<tr>
<th>Reporter</th>
<td>
Emilyaxe
</td>
</tr>
</table>
<pre>
git version: 59fd2878fce47
system: `Ubuntu 18.04.6 LTS`
## Description:
I am experiencing an inconsistent result when executing the same MLIR program with and without `--test-affine-parametric-tile`.
## Steps to Reproduce:
### 1. **MLIR Program (a.mlir)**:
a.mlir:
```
module {
func.func private @printMemrefI32(tensor<*xi32>)
func.func private @printMemrefF32(tensor<*xf32>)
func.func @entry(%arg0: index) -> () {
%1 = "tosa.const"() <{values = dense<7> : tensor<1x4x5xi32>}> : () -> tensor<1x4x5xi32>
%5 = "tosa.const"() <{values = dense<-2522> : tensor<1x4x5xi32>}> : () -> tensor<1x4x5xi32>
%16 = tosa.clamp %5 { max_val = 2 : i32, min_val = 0 : i32} : (tensor<1x4x5xi32>) -> tensor<1x4x5xi32>
%17 = tosa.logical_right_shift %1, %16 : (tensor<1x4x5xi32>, tensor<1x4x5xi32>) -> tensor<1x4x5xi32>
%19 = tosa.reduce_product %17 {axis = 1 : i32} : (tensor<1x4x5xi32>) -> tensor<1x1x5xi32>
%cast_4 = tensor.cast %19 : tensor<1x1x5xi32> to tensor<*xi32>
call @printMemrefI32(%cast_4) : (tensor<*xi32>) -> ()
return
}
func.func @main() {
%idx0 = index.constant 0
call @entry(%idx0) : (index) -> ()
return
}
}
```
### 2. **Command to Run without `--test-affine-parametric-tile` :**
```
/data/szy/MLIR/llvm-release/llvm-project/build/bin/mlir-opt /data/szy/workspace/mlir-inconsistent/a.mlir \
-pass-pipeline="builtin.module(func.func(tosa-to-linalg-named,tosa-to-linalg))" \
| /data/szy/MLIR/llvm-release/llvm-project/build/bin/mlir-opt -tosa-to-arith --convert-elementwise-to-linalg -convert-linalg-to-loops \
-one-shot-bufferize="bufferize-function-boundaries" --linalg-fold-unit-extent-dims -convert-linalg-to-affine-loops \
| /data/szy/MLIR/llvm-release/llvm-project/build/bin/mlir-opt -pass-pipeline="builtin.module(func.func(affine-loop-unroll,affine-loop-unroll))" \
| /data/szy/MLIR/llvm-release/llvm-project/build/bin/mlir-opt --test-constant-fold -convert-index-to-llvm -convert-arith-to-llvm \
--expand-strided-metadata -lower-affine -convert-scf-to-cf -finalize-memref-to-llvm -convert-arith-to-llvm -convert-cf-to-llvm \
-convert-func-to-llvm -convert-math-to-llvm -reconcile-unrealized-casts \
| timeout 10 /data/szy/MLIR/llvm-release/llvm-project/build/bin/mlir-cpu-runner -e main -entry-point-result=void \
--shared-libs=/data/szy/MLIR/llvm-release/llvm-project/build/lib/libmlir_c_runner_utils.so \
--shared-libs=/data/szy/MLIR/llvm-release/llvm-project/build/lib/libmlir_runner_utils.so \
--shared-libs=/data/szy/MLIR/llvm-release/llvm-project/build/lib/libmlir_async_runtime.so
```
### 3. **Output without `--test-affine-parametric-tile` :**:
```
[[[2401, 2401, 2401, 2401, 2401]]]
```
### 4. **Command to Run with `--test-affine-parametric-tile` :**
```
/data/szy/MLIR/llvm-release/llvm-project/build/bin/mlir-opt /data/szy/workspace/mlir-inconsistent/a.mlir \
-pass-pipeline="builtin.module(func.func(tosa-to-linalg-named,tosa-to-linalg))" \
| /data/szy/MLIR/llvm-release/llvm-project/build/bin/mlir-opt -tosa-to-arith --convert-elementwise-to-linalg -convert-linalg-to-loops \
-one-shot-bufferize="bufferize-function-boundaries" --linalg-fold-unit-extent-dims -convert-linalg-to-affine-loops \
| /data/szy/MLIR/llvm-release/llvm-project/build/bin/mlir-opt -pass-pipeline="builtin.module(func.func(affine-loop-unroll,affine-loop-unroll))" \
| /data/szy/MLIR/llvm-release/llvm-project/build/bin/mlir-opt --test-constant-fold -convert-index-to-llvm -convert-arith-to-llvm \
--expand-strided-metadata --test-affine-parametric-tile -lower-affine -convert-scf-to-cf -finalize-memref-to-llvm -convert-arith-to-llvm \ -convert-cf-to-llvm -convert-func-to-llvm -convert-math-to-llvm -reconcile-unrealized-casts \
| timeout 10 /data/szy/MLIR/llvm-release/llvm-project/build/bin/mlir-cpu-runner -e main -entry-point-result=void \
--shared-libs=/data/szy/MLIR/llvm-release/llvm-project/build/lib/libmlir_c_runner_utils.so \
--shared-libs=/data/szy/MLIR/llvm-release/llvm-project/build/lib/libmlir_runner_utils.so \
--shared-libs=/data/szy/MLIR/llvm-release/llvm-project/build/lib/libmlir_async_runtime.so
```
### 5. **Output with `--test-affine-parametric-tile` :**
```
[[[712284096, 22092, 0, 0, 712725616]]]
```
### 6. **Analysis for this case :**
I debug this issue and find the faulty pass is `--test-affine-parametric-tile` pass
The input IR (ir before running the `--test-affine-parametric-tile` ) can be found in [input.txt](https://github.com/user-attachments/files/19225730/input.txt)
The output IR (ir after running the `--test-affine-parametric-tile` ) can be found in [output.txt](https://github.com/user-attachments/files/19225735/output.txt)
Please change file from .txt to .mlir
The `--test-affine-parametric-tile` pass utilizes SSA values as tiling parameters (tile sizes); however, it does not account for the case where a tile size is 0, which can lead to undefined behavior.
</pre>
<img width="1" height="1" alt="" src="http://email.email.llvm.org/o/eJzsWV9vozoW_zTOy5EjMBDIQx7S5EaqdK_2amb2uXLgELxrbGSbNJ1Pv7KBNG1ze7OdmdU-VEIJwcfn_M7fHwFurTgoxBXJ7ki2nfHeNdqsfmuFfOInnO119bQ6CAdHNFZoRZI1ZMu6YkVe1CWmOYnWJFrbJ-uw9YtkEf1z3yvXQ1zMo3S-gN-_fSWLaJAjLCEsgS3a0ojODQpJtL4H3gKeOjQCVSnUAbgCoUqtrLAOlQODtpcOHhtUgCcse-elXINgeYvwx-_3X6Az-mB4C4_CNcBVFU507zwoSh1aR3ldC4W044a36IwoqRMSySKaw4jwDPKrw86C0_AFO6OrvsQB61nCC8VzIGxN2DoA-HMEQFjB560UhrDlsDxsHS_6MEVrH5Rw-B-trnqJQPI7Eq2h7lU59x_QGXHkDoGkUWeEcn9ga7C-TxhhhUNltSHJhrD1SSSMJL95e9Ea_lbB7q2C-i8UkDRC5cwTYQVhGTeHyOMXqsITYUugJPkNwtpyQk9YFgNJtkAYc9ryuU-jI4xNYsmG5HdHLnu0Qa5CZZEkmzzoStZwBhaf0lM2-ZZvp_VRUbB9Xdb7AB5J9iEklGWM_QI08SKYGbBI3nYjxPwOWn56OHIZ1lnQ6zezDUAr1HklOq_k28n4dZu3IsqfEUl9ECWXD0YcGvdgG1G7IOJRTODfs7j5C2O3Qlk-QzHoG-5haDw3Ac3v-EkMmYp_JBDxFeslt-4hHRAEybm_csa1vr7fz4drfTioLbmUV1v3bG-owpfoX7TzZYN5rQZdbxQEA74G3zRry4V62Y-Df6I6RcG70LpDJ3DlIHqF9aLZ_ZYLgNd6ftw8oLoANX5ejrhhNEyDk02Dc6Pb1s9qP2h7dR7Zt8xsD2wcr8NYfmGO7SruOGE7-_2JsJ0f0ITtpDy21KBEbnH62Rn9LywdYbt9L2Tlv30Md35YU935Inip61Gbf9uOlzgJXVIVYbthzgPJNiRa045bSzvRoRQKSbIljHk7Tqj5MPYJK84p9IWgLadOUykUlweqeIsVYZuXlwOzLAljoxWSb97A_BGX6WSOG8-mQGmp1RGNoyixReUehcVnOADn9RG2X9K6syM-oFohtY12dN_XNRrx_RyL8Sf1_vt7ArrXvaq4EWi9g3RSWWtZ0V4JR_HkI00r0dprhseKGez_qgD9t2m9AEV7ZbSUhG2uXfzFiR2aaur_ENSL7IUmD8mTxxYuFkIdnBfG0qZ46riqqHVGVFjRFh33SP1GqR_RjKm4VGTL2qspa6C1z5jPfBtm49-bvVgo69dgphUf72s7Wn6hiRostSqFRB93DDAq6oeyvQi8Ey36aRRHPy0HZddT0yuFBiiCn9ZAw8ylnRbK0eFGlyTboxbVOc624QYrKsXehlL7ASxS7IdPD-ehfBjAPPROSDu3-teb_J8b5PZJBT99OudWX2GmZ2JKJmL6R-86T0QfZKTkKin5f1n-YGkUbqsA4LbTbDse72FP3yHVjzHqJ6l-kuonqf6fkOq73fuGcq8Q7gcYl2Sbq5z7ybafbPsT2DZ7xbaBqX74r9_EsnnMWJFGy8XEpCxasvE8evWdxyxn2SJe3Ea1iwn4WnH5ZIWFWhtwjbBQcouv8d1Dhfv-MAgIa3sMDyhr4Um6Qah5L90T-PEHwt4UgSBMovW3BkEoH7z7L-FvuoE91tog-NRPT0lviilbQskV7BFqTxYgFJDsLiifu5PzMWFF41xng3s7wnYH4Zp-Py91S9iut378OMfLxrOZJWxXC-kJZxcvGcvyJCJs96wvPD_w-PWQ_bMDvHZofhb-QflPciAjbHehMHjwZ-gPKBuuDgh-A9RGt-Al_C3Y8NQ31MG3G10JheB7VnxHC1-_rmF8Sskt-KvqAOMuNDY8PfJWrZf2mJI7aPQjHtH46hYOKo0WlHbAy1L3yo3VikOxPjZoEDiclfgaDI3x2IiyCUGVyMP9ZK8q9Kgr2GPDj0Kb-dQbMcnu7lt-wI_EmVuL4QTZokyLrKAxqxY0LXBBi6xAWi2Kot4vcZ_G6RD3t8esWiXVMlnyGa7iPI2LbFGk0axZ5XFZFIt0EZU58hrTOksY40lcLaOq4Hk0EysWsSxK4iRaRjmL5rGXLIo0qpL9siz3JI2w5ULO_RCca3OYhTZexUkcR9lM8j1KG16jMDY8-mck287MKgzNfX-wJI2ksM4-a3DCyfDqJWzItnB_-cpjbIpXrzzevum4oZ5mvZGrd_LhEb2d78HBUPuDj8cV-08AAAD__7I8Nms">