[XPU] fix ep_tp all2all ci (#4876)

This commit is contained in:
zhupengyang
2025-11-07 16:37:23 +08:00
committed by GitHub
parent cba185f1fe
commit 6b5ae9ffea

View File

@@ -352,7 +352,7 @@ export BKCL_RDMA_VERBS=1
export enable_expert_parallel=1
export enable_tensor_parallel=1
export EP_TP_SPLIT_MODE=1
export FD_EP_TP_STRATEGY=all_to_all
python -m pytest -s --timeout=600 tests/ci_use/XPU_45T/run_ep.py
ep_exit_code=$?
@@ -366,7 +366,7 @@ unset XSHMEM_QP_NUM_PER_RANK
unset BKCL_RDMA_VERBS
unset enable_expert_parallel
unset enable_tensor_parallel
unset EP_TP_SPLIT_MODE
unset FD_EP_TP_STRATEGY
stop_processes
if [ ${ep_exit_code} -ne 0 ]; then