apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2026-04-23 00:17:25 +08:00
Code Issues Actions 19 Packages Projects Releases Wiki Activity

[BugFix] fix compile error in sm89 (#5809)

Browse Source
This commit is contained in:
周周周
2025-12-29 16:55:52 +08:00
committed by GitHub
parent eb782a0225
commit a3f0696e35
2 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
custom_ops/setup_ops.py
+1
View File
@@ -493,6 +493,7 @@ elif paddle.is_compiled_with_cuda():
# Hopper optimized mla
sources += find_end_files("gpu_ops/mla_attn", ".cu")
sources += ["gpu_ops/flash_mask_attn/flash_mask_attn.cu"]
cc_compile_args += ["-DENABLE_FLASH_MASK_ATTENTION"]
sources += find_end_files("gpu_ops/moba_attn/moba_decoder_attn/", ".cu")
sources += find_end_files("gpu_ops/moba_attn/moba_encoder_attn/", ".cu")
sources += find_end_files("gpu_ops/moba_attn/moba_process/", ".cu")