tree cd326823a26f9df30f3795a7bd74b8a5ded6f93e
parent 44808e143533e694644da13b75a8f8358ee289cf
author Stanley Winata <68087699+raikonenfnu@users.noreply.github.com> 1720827039 -0700
committer GitHub <noreply@github.com> 1720827039 -0700
gpgsig -----BEGIN PGP SIGNATURE-----
 
 wsFcBAABCAAQBQJmkbyfCRC1aQ7uu5UhlAAAdScQAF1N3FRBfXrEdG0e5ZH1HJ2x
 AfFVdtkV9KQE1YM4FdSomhW0PJZgt6ueabBIpwc+FTJoBBaZ1jYCs8IeD24IVwH4
 xXx+Kf1nCwX2s/lOQgRwl1C54dk+dhIkaRV+xaVU2W5MZ/LlP6rH26bk/DcRWcBl
 Xh7szc1SMQ7NCu3kseVKo3HlsGz0atYSjgha/Yldm5riIJlSSQEE5OHpBLIfZhQB
 ximeOxhq3lpDOYNebjMRw2pIpIBKK/DLrgBlz0xQD9ztpFjAzQy2hpk/ub0GV07P
 O8ajs/aWmuYm8Hbx5sw0lWPZWarF/7cN3fAsk5o3VkHFyGQiooUXfS/T4h4RT/OS
 YY4Jhz57VzlxrxdZg8jVGiL/5rBAFssmWxMQJQgPRJ13Dc4C1JOZORU1pX6nJg44
 2Y9m2WuDQev4Neo8htDDxtqcgcnDtrMbZPGIbVqKWOtBvPo7rAQ92gWPlXHYJZzO
 QbpyQF0drfyCNHqmtm0CthlNAJjrtYIokf+z8xjlJnm/9JAMcDwWouKwb7nQzgl5
 rBlsvFj3rhGP6goW5NM5FxVBZJrrqar44M1GiveV395n3CzucEQTCsDlaR8+sEo+
 tQ+lhHit3LEv0sRT8ZXlvIcoJID7cZDBK+1knEwJkn3es9Syhona8sMa21NXKPG5
 cbAQTfqr5eRiJUMORrGo
 =UyAr
 -----END PGP SIGNATURE-----
 

[LLVMGPU] Support CastTypeToFitMMA on TransformDialect script.  (#17884)

Previously CastTypeToFitMMA relies on the `mma_schedule` attribute on
the function's translationInfo to obtain information about
`iree.amdgpu.mma`(intrnisic selected).

While this is fine for C++ pipeline, the IR generated from
TransformDialect script do not have such information. Instead IR
generated in TD script typically annotate the
`iree.amdgpu.mma`(intrnisic selected) directly on the
vector.contractOps.

This is a crucial part of enabling performant the latest attention
compilation pipeline (with online attn + transpose fusion) which is
based on TD scripts.

---------

Co-authored-by: Kunwar Grover <groverkss@gmail.com>