[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