We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent b157d5f commit 34fcbe4Copy full SHA for 34fcbe4
csrc/setup_cuda.py
@@ -226,14 +226,13 @@ def get_gencode_flags():
226
"gpu/fp8_gemm_with_cutlass/fp8_fp8_half_block_gemm.cu",
227
"gpu/fp8_gemm_with_cutlass/fp8_fp8_half_gemm_ptr_scale.cu",
228
]
229
- sources += find_end_files("./gpu/mla_attn", ".cu")
230
+ sources += find_end_files("./gpu/mla_attn", ".cu")
231
sources += [
232
"gpu/sparse/sparse_compressor.cu",
233
"gpu/sparse/sparse_mm.cu",
234
"gpu/sparse/sparse_mm_impl.cu",
235
236
-
237
238
ops_name = f"paddlenlp_ops_{sm_version}" if sm_version != 0 else "paddlenlp_ops"
239
0 commit comments