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

    <tr>
        <th>Summary</th>
        <td>
            [MLIR] Conversion to LLVM fails for AMX tile store intrinsic with BlockArgument
        </td>
    </tr>

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

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

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

<pre>
    Here is a small reproducer:
```
llvm.func @amx_phi(%copy: i1, %src: !llvm.ptr, %dst: !llvm.ptr) {
  %cst_16 = llvm.mlir.constant(16 : i16) : i16
  %cst_32 = llvm.mlir.constant(32 : i64) : i64
  %0 = "amx.tilezero"(%cst_16, %cst_16) : (i16, i16) -> !llvm.array<16 x vector<16xbf16>>
  llvm.br ^bb1(%0 : !llvm.array<16 x vector<16xbf16>>)
 ^bb1(%1: !llvm.array<16 x vector<16xbf16>>):
 "amx.tilestored64"(%cst_16, %cst_16, %dst, %cst_32, %1) : (i16, i16, !llvm.ptr, i64, !llvm.array<16 x vector<16xbf16>>) -> ()
 llvm.return
}
```
Translation to LLVM fails with the following error:
```
$ ./bin/mlir-translate --mlir-to-llvmir amx-phi.mlir
mlir-translate: /localdisk/ilyaenko/llvm-project/llvm/lib/IR/Instructions.cpp:685: void llvm::CallInst::init(llvm::FunctionType*, llvm::Value*, llvm::ArrayRef<llvm::Value*>, llvm::ArrayRef<llvm::OperandBundleDefT
<llvm::Value*> >, const llvm::Twine&): Assertion `(i >= FTy->getNumParams() || FTy->getParamType(i) == Args[i]->getType()) && "Calling a function with a bad signature!"' failed.
```
It looks like the store can be successfully translated only when its source tile is produced by another already translated AMX intrinsic call that provides `x86_amx` result instead of the original `[16 x <16 x float>]`. This makes it impossible to lower a loop with `amx.tdp*` call keeping an accumulator in a tile and storing it only after the loop.

I wonder if this can be fixed in MLIR code, e.g. by adding some explicit casts. If this has to be fixed in the translator, I'd appreciate any guidance on how it can be fixed.
</pre>
<img width="1px" height="1px" alt="" src="http://email.email.llvm.org/o/eJycVl1v47gO_TXKC9HAlmPHechDmkxwC8zcezEoBvs2kCU65laWDElOk_31C8lum-5M9wsw2sgkj8nDQ9rCezoZxC0r71l5WIgxdNZtCc2TPZPsFo1V1-1_0CGQBwG-F1qDw8FZNUp0rNix7MCyHauy-UpHrc_9sh2NBLbKRH_5PnTEeM14Ke1wZcUOKGd8D4yX3sl4ZjxPQUNws0H58INhA2x9Pz0Coo_04XteASsOkJx6TW4prfFBmMB4nWzxYVWKnX--jy_4x_HJtgOqVq_x1eomPkuhjHPRX5aBNP6GzjLO51pTdnM5L4cJhvGaJtOc2x0rPr3WKpwTV1bs8woucEYZrEunS9PmFSs-xWtOIvk3Dlj5qWny6bkZ3BL3d8D4Zsa7hcn_BcqLHm458cE6VJHCP-Pltedvtws-H_IPaNv_UTapUft_mPQL9_UbDSnaYRidmfW9PvxU6I9OGK9FIGsgWPj8-dsXaAVpD88UOggdQmu1ts9kToDO2Q9HhvEVLBk_NmQYP0Yh3oUZHOHubrph72Jm5ED0l7uhoyTYKf59xETWUVsptCL_xPiR9FXEuY639bm_G5z9FWWYj_EfNYwfH77GP8YHN8pYll_KYWDFrqrLCHq2pBI9sY5itxdaR-fpRIbi1LyZj6NJII_XARnfxd68Gb8JPf54dxd79hVbVux_4hob9pfe_xvQCaPuR6M0HrB9nBn-ABFm1DT3N9iPz2SQ8WrSNey8R5caHXvGa0pxxQGOj9eooBOG_479_4UTvZ_EBGy9Z-v9jUOyTmTUNIn6ECF27uRZeU-sPMyes1N8dnSLWVRxqCLhUUsC2pnbSWkCGqEgbnMRRoeM52nc1kmNqJY_1dxDAG3tkwdNT5jEmmYVpDDQIPhRSvS-HbW-wqu0FFijr_DcoQEKHrwdnUSIkx5fEvOrQUFzBWFs6NCB0A6Feoex-_ILkAmOjCcJMr5WQidCDD-TQh9JvtTVd9FfWJWBQz_qAGR8QKHAtilb6-hERujUkfI-jfnLuLfaihA7VB5YlS3hsSMPvXhCDxSA-sF6T43GOLbaPscsIxnDRCersrS-1BA1UmVThk-IQyLfgJBy7EctgnVABsRUvzAqURidKExEiTagS-lG-JdGTPzDszUKHVCsh_wL8S1dUEXYL58fvoK0CqM-cXlaJlaVivje9gh4GTRJCiCFD34JDzNQJ3ws7BYrZvDSAJu25QPjawViGBxKimtGmCucRlLCSARroLPPkLDfslou1LZQm2IjFrjN17yqq3VRbxbdVmx4jkphKZqKVwLLom02RVEo2WzKLF8taMszvso2POM5L7J6WddFqcqmbnIUm6xWbJVhL0gv0_q17rQg70fc5tlmVecLLRrUPn2qcJ4WH-fxo8Vt0z5rxpNnq0xTJOIVIVDQ6fMmUsnKA-ytOaPzPy7s1rqkytTIaQ7eBJpEca-tfNq509ijCYvR6W0XwuDjtuBHxo8nCt3YLKXtb5bq-1WbCvKMH-eazlv-ewAAAP__sIPmtw">