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 21 Packages Projects Releases Wiki Activity

[Feature] Support noaux for eplb (#5143)

Browse Source 
* support noaux eplb

* noaux_eplb

* noaux_eplb

* noaux_eplb
This commit is contained in:
xiaoxiaohehe001
2025-11-21 14:10:32 +08:00
committed by GitHub
parent e70e2279ce
commit 6ca2651995
8 changed files with 616 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files
custom_ops/setup_ops.py
+2
View File
@@ -301,6 +301,7 @@ elif paddle.is_compiled_with_cuda():
"gpu_ops/get_position_ids_and_mask_encoder_batch.cu",
"gpu_ops/fused_rotary_position_encoding.cu",
"gpu_ops/noaux_tc.cu",
"gpu_ops/noaux_tc_redundant.cu",
"gpu_ops/custom_all_reduce/all_reduce.cu",
"gpu_ops/merge_prefill_decode_output.cu",
"gpu_ops/limit_thinking_content_length_v1.cu",
@@ -614,6 +615,7 @@ elif paddle.device.is_compiled_with_custom_device("metax_gpu"):
"gpu_ops/share_external_data.cu",
"gpu_ops/recover_decode_task.cu",
"gpu_ops/noaux_tc.cu",
"gpu_ops/noaux_tc_redundant.cu",
"gpu_ops/fused_rotary_position_encoding.cu",
"gpu_ops/text_image_gather_scatter.cu",
"gpu_ops/text_image_index_out.cu",