[Mlir-commits] [mlir] [mlir][nvvm] Verify kind-specific target requirements of tcgen05.mma ops (PR #235628)

[llvmlistbot at llvm.org]

Vaisman wrote:

> > kind::i8 needs sm_100a or sm_110a.
> 
> > Sparse block-scaled mxf4 and mxf4nvf4 need sm_100a, sm_103a, sm_107a or sm_110a.
> 
> Shouldn't these also include sm_101a? The NVPTX backend defines sm_101 and sm_110 as aliases.

Yes, thanks! sm_101* are aliases of sm_110* in NVPTX. I added sm_101a to both lists, and the check now canonicalizes these aliases in both the target and required SM versions before comparing them. This also fixes scale_input_d, which was wrongly accepted on sm_101a/f. Added positive tests for i8 and sparse mxf4 on sm_101a, and negative scale_input_d tests on sm_101a and sm_101f.

https://github.com/llvm/llvm-project/pull/235628