apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-10-05 00:33:03 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

[MetaxGPU] adapt to the latest fastdeploy on metax gpu (#3492)

Browse Source
This commit is contained in:
Kane2011
2025-08-25 17:44:20 +08:00
committed by GitHub
parent c13c904971
commit 2ae7ab28d2
8 changed files with 338 additions and 115 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
custom_ops/setup_ops.py
View File

@@ -591,6 +591,12 @@ elif paddle.device.is_compiled_with_custom_device("metax_gpu"):
if not os.listdir(json_dir):
raise ValueError("Git clone nlohmann_json failed!")
sources = [
"gpu_ops/update_inputs_v1.cu",
"gpu_ops/save_with_output_msg.cc",
"gpu_ops/get_output.cc",
"gpu_ops/get_output_msg_with_topk.cc",
"gpu_ops/save_output_msg_with_topk.cc",
"gpu_ops/transfer_output.cc",
"gpu_ops/save_with_output.cc",
"gpu_ops/set_mask_value.cu",
"gpu_ops/set_value_by_flags.cu",