[Metax] adapt to the latest develop (#6282)

This commit is contained in:
xiaozude
2026-01-30 15:21:20 +08:00
committed by GitHub
parent 18ebce9dec
commit 030647521a
14 changed files with 754 additions and 370 deletions
+2
View File
@@ -643,6 +643,8 @@ elif paddle.device.is_compiled_with_custom_device("metax_gpu"):
"gpu_ops/ipc_sent_key_value_cache_by_remote_ptr.cu",
"gpu_ops/unset_data_ipc.cu",
"gpu_ops/swap_cache_batch.cu",
"gpu_ops/gelu_tanh.cu",
"gpu_ops/set_stop.cu",
"metax_ops/moe_dispatch.cu",
"metax_ops/moe_ffn.cu",
"metax_ops/moe_reduce.cu",