diff --git a/scripts/run_ci_xpu.sh b/scripts/run_ci_xpu.sh index 357870c1fbd..d96d9db6a7c 100644 --- a/scripts/run_ci_xpu.sh +++ b/scripts/run_ci_xpu.sh @@ -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