支持nmz fp8

c4412432 · zhanghj2 · 26d2ab19 · c4412432
Commit c4412432 authored Feb 06, 2026 by zhanghj2
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 0 deletions

setup.py setup.py +3 -0

No files found.
--- a/setup.py
+++ b/setup.py
@@ -57,6 +57,9 @@ ext_modules.append(
            "csrc/sm90/decode/dense/instantiations/fp16.cu",
            "csrc/sm90/decode/dense/instantiations/bf16.cu",

+            ## sm90 dense qkvfp8 decode
+            "csrc/sm90/decode/dense_qkvfp8/instantiations/fp8e4m3.cu",
+            
            # # sm90 sparse decode
            "csrc/sm90/decode/sparse_fp8/instantiations/model1_persistent_h16.cu",
            "csrc/sm90/decode/sparse_fp8/instantiations/model1_persistent_h64.cu",