diff --git a/custom_ops/xpu_ops/download_dependencies.sh b/custom_ops/xpu_ops/download_dependencies.sh index e5e3ddcbd..7e5ff7586 100644 --- a/custom_ops/xpu_ops/download_dependencies.sh +++ b/custom_ops/xpu_ops/download_dependencies.sh @@ -12,8 +12,8 @@ rm -rf "$THIRDPARTY_DIR" mkdir -p "$THIRDPARTY_DIR" || exit 1 if [ "$1" == "stable" ]; then - version_xvllm="20250902" - version_xtdk="3.4.40.1" + version_xvllm="20251017" + version_xtdk="3.4.0.1" else version_xvllm="latest" version_xtdk="latest" diff --git a/scripts/run_ci_xpu.sh b/scripts/run_ci_xpu.sh index 9afd81249..1b339609e 100644 --- a/scripts/run_ci_xpu.sh +++ b/scripts/run_ci_xpu.sh @@ -32,6 +32,7 @@ python -m pip install openai -U python -m pip uninstall -y triton python -m pip install triton==3.3.0 python -m pip install pytest +python -m pip install pytest-timeout unset http_proxy unset https_proxy unset no_proxy @@ -197,7 +198,7 @@ cd xDeepEP bash build.sh cd - -python tests/ci_use/XPU_45T/run_ep.py +python -m pytest -s --timeout=300 tests/ci_use/XPU_45T/run_ep.py ep_exit_code=$? if [ ${ep_exit_code} -ne 0 ]; then