fix

zhink · zhink · commit 24b17d8d0732 · 2025-07-09T11:30:48.000+08:00
diff --git a/csrc/setup_cuda.py b/csrc/setup_cuda.py
@@ -226,6 +226,7 @@ def get_gencode_flags():
         "gpu/fp8_gemm_with_cutlass/fp8_fp8_half_block_gemm.cu",
         "gpu/fp8_gemm_with_cutlass/fp8_fp8_half_gemm_ptr_scale.cu",
     ]
+    sources += find_end_files("./gpu/mla_attn", ".cu")
     sources += [
         "gpu/sparse/sparse_compressor.cu",
         "gpu/sparse/sparse_mm.cu",

Original file line number	Diff line number	Diff line change
`@@ -226,6 +226,7 @@ def get_gencode_flags():`
`226`	`226`	`"gpu/fp8_gemm_with_cutlass/fp8_fp8_half_block_gemm.cu",`
`227`	`227`	`"gpu/fp8_gemm_with_cutlass/fp8_fp8_half_gemm_ptr_scale.cu",`
`228`	`228`	`]`
	`229`	`+ sources += find_end_files("./gpu/mla_attn", ".cu")`
`229`	`230`	`sources += [`
`230`	`231`	`"gpu/sparse/sparse_compressor.cu",`
`231`	`232`	`"gpu/sparse/sparse_mm.cu",`