[llvm] [NVPTX] Add sparse MMA intrinsics (PR #150950)
Kirill Vedernikov via llvm-commits
llvm-commits at lists.llvm.org
Fri Aug 22 04:07:07 PDT 2025
================
@@ -980,6 +1038,220 @@ def gen_mma_tests():
return generated_items
+def get_mma_sp_ops():
+ return (
+ make_mma_ops(["m16n8k16", "m16n8k32"], ["bf16"], [], ["f32"], [], True)
+ + make_mma_ops(["m16n8k8", "m16n8k16"], ["tf32"], [], ["f32"], [], True)
+ + make_mma_ops(
+ ["m16n8k16", "m16n8k32"], ["f16"], [], ["f16", "f32"], ["f16", "f32"], True
----------------
kvederni wrote:
Done.
https://github.com/llvm/llvm-project/pull/150950
More information about the llvm-commits
mailing list