fix

zhink · zhink · commit c3d988383ffe · 2025-05-12T11:33:59.000+08:00
diff --git a/csrc/setup_cuda.py b/csrc/setup_cuda.py
@@ -226,14 +226,12 @@ 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")
-
-if cc >= 90 and cuda_version >= 12.2:
     sources += [
         "gpu/sparse/sparse_compressor.cu",
         "gpu/sparse/sparse_mm.cu",
         "gpu/sparse/sparse_mm_impl.cu",
     ]
+    
 
 ops_name = f"paddlenlp_ops_{sm_version}" if sm_version != 0 else "paddlenlp_ops"
 

Original file line number	Diff line number	Diff line change
`@@ -226,14 +226,12 @@ 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")`
`230`		`-`
`231`		`-if cc >= 90 and cuda_version >= 12.2:`
`232`	`229`	`sources += [`
`233`	`230`	`"gpu/sparse/sparse_compressor.cu",`
`234`	`231`	`"gpu/sparse/sparse_mm.cu",`
`235`	`232`	`"gpu/sparse/sparse_mm_impl.cu",`
`236`	`233`	`]`
	`234`	`+`
`237`	`235`
`238`	`236`	`ops_name = f"paddlenlp_ops_{sm_version}" if sm_version != 0 else "paddlenlp_ops"`
`239`	`237`