DeepSeek/FlashMLA
1
0
Fork 0
mirror of https://github.com/deepseek-ai/FlashMLA synced 2025-06-26 18:15:54 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fp8 shared mem

Browse Source
This commit is contained in:
chenhongmin.will
2025-02-25 11:08:28 +08:00
parent b67a18f850
commit fed0499301
2 changed files with 9 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
setup.py
View File

@@ -37,7 +37,7 @@ ext_modules.append(
sources=[
"csrc/flash_api.cpp",
"csrc/flash_fwd_mla_bf16_sm90.cu",
"csrc/flash_fwd_mla_fp8_sm90.cu",
#"csrc/flash_fwd_mla_fp8_sm90.cu",
],
extra_compile_args={
"cxx": cxx_args,