mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2026-04-23 00:17:25 +08:00
YuBaoku
1269eda2f9
* [CI] Ensure container cleanup after job to avoid resource leakage * [CI] Use prebuilt wheels to install xgrammar==0.1.19 and torch==2.6.0
85 lines
3.5 KiB
Bash
85 lines
3.5 KiB
Bash
#!/bin/bash
|
|
DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
|
echo "$DIR"
|
|
|
|
# python -m pip install --pre paddlepaddle-gpu -i https://www.paddlepaddle.org.cn/packages/nightly/cu126/
|
|
python -m pip config set global.index-url https://mirrors.tuna.tsinghua.edu.cn/pypi/web/simple
|
|
|
|
python -m pip install -r requirements.txt
|
|
python -m pip install jsonschema aistudio_sdk==0.3.5
|
|
# Use prebuilt wheel files to install xgrammar==0.1.19 and torch==2.6.0 specifically for the CI environment
|
|
python -m pip install \
|
|
https://paddle-qa.bj.bcebos.com/FastDeploy/torch-2.6.0-cp310-cp310-manylinux1_x86_64.whl \
|
|
https://paddle-qa.bj.bcebos.com/FastDeploy/triton-3.2.0-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl \
|
|
https://paddle-qa.bj.bcebos.com/FastDeploy/xgrammar-0.1.19-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
|
|
|
|
failed_files=()
|
|
run_path="$DIR/../tests/ci_use/"
|
|
|
|
# load all test files
|
|
for subdir in "$run_path"*/; do
|
|
if [ -d "$subdir" ]; then
|
|
pushd "$subdir" > /dev/null || continue # into test dir or continue
|
|
|
|
# search for test_*.py files
|
|
for file in test_*.py; do
|
|
if [ -f "$file" ]; then
|
|
echo "============================================================"
|
|
echo "Running pytest on $(realpath "$file")"
|
|
echo "------------------------------------------------------------"
|
|
|
|
set +e
|
|
timeout 600 python -m pytest --disable-warnings -sv "$file"
|
|
exit_code=$?
|
|
set -e
|
|
ps -ef | grep "${FD_CACHE_QUEUE_PORT}" | grep -v grep | awk '{print $2}' | xargs -r kill -9
|
|
ps -ef | grep "${FD_ENGINE_QUEUE_PORT}" | grep -v grep | awk '{print $2}' | xargs -r kill -9
|
|
|
|
if [ $exit_code -ne 0 ]; then
|
|
if [ -d "${subdir%/}/log" ]; then
|
|
echo ">>> grep error in ${subdir%/}/log/"
|
|
grep -Rni --color=auto "error" "${subdir%/}/log/" || true
|
|
else
|
|
echo "${subdir%/}/log directory not found"
|
|
fi
|
|
|
|
if [ -f "${subdir%/}/log/workerlog.0" ]; then
|
|
echo "---------------- log/workerlog.0 -------------------"
|
|
cat "${subdir%/}/log/workerlog.0"
|
|
echo "----------------------------------------------------"
|
|
fi
|
|
|
|
if [ -f "${subdir%/}/server.log" ]; then
|
|
echo "---------------- server.log ----------------"
|
|
cat "${subdir%/}/server.log"
|
|
echo "--------------------------------------------"
|
|
fi
|
|
|
|
if [ "$exit_code" -eq 1 ] || [ "$exit_code" -eq 124 ]; then
|
|
echo "[ERROR] $file 起服务或执行异常，exit_code=$exit_code"
|
|
if [ "$exit_code" -eq 124 ]; then
|
|
echo "[TIMEOUT] $file 脚本执行超过 10 分钟, 任务超时退出！"
|
|
fi
|
|
fi
|
|
|
|
failed_files+=("$subdir$file")
|
|
exit 1
|
|
fi
|
|
echo "------------------------------------------------------------"
|
|
fi
|
|
done
|
|
popd > /dev/null # back to test dir
|
|
fi
|
|
done
|
|
|
|
if [ ${#failed_files[@]} -gt 0 ]; then
|
|
echo "The following tests failed:"
|
|
for f in "${failed_files[@]}"; do
|
|
echo "$f"
|
|
done
|
|
exit 1
|
|
else
|
|
echo "All tests passed!"
|
|
exit 0
|
|
fi
|