Files
paddlepaddle--paddle/ci/formers_test.sh
T
2026-07-13 12:40:42 +08:00

69 lines
2.2 KiB
Bash

# Copyright (c) 2025 PaddlePaddle Authors. All Rights Reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
function formers_api() {
cd /workspace/PaddleFormers && git config --global --add safe.directory $PWD
echo "Check whether the local model file exists:"
ls -l ./models
timeout 30m bash scripts/unit_test/ci_unittest.sh ${paddle_whl} false ${PYTEST_EXECUTE_FLAG_FILE} ${BRANCH}
}
function formers_models() {
rm -rf /root/.cache/aistudio/
cd /workspace/PaddleFormers && git config --global --add safe.directory $PWD
echo "Check whether the local model file exists:"
ls -l ./models
timeout 30m bash scripts/regression/ci_model_unittest.sh ${paddle_whl} ${BRANCH}
}
function formers_test() {
python /workspace/tools/get_pr_title.py skip_distribute_test && CINN_OR_BUAA_PR=1
if [[ "${CINN_OR_BUAA_PR}" = "1" ]];then
echo "PR's title with 'CINN' or 'BUAA', skip the run distribute ci test !"
exit 0
fi
echo "::group::Start formers api tests"
formers_api
echo "End api tests"
echo "::endgroup::"
echo "::group::Start formers models tests"
formers_models
echo "End models tests"
echo "::endgroup::"
}
set -e
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/lib64/:/usr/local/lib/
PATH=/usr/local/bin:${PATH}
ln -sf $(which python3.10) /usr/local/bin/python
ln -sf $(which pip3.10) /usr/local/bin/pip
echo "Downloading PaddleFormers.tar.gz..."
wget -q https://paddleformers.bj.bcebos.com/wheels/PaddleFormers.tar.gz
tar xf PaddleFormers.tar.gz
echo "Extracting PaddleFormers.tar.gz..."
cd PaddleFormers
cp -r ${CFS_DIR}/models ./models
echo "::group::Install paddle dependencies"
pip config set global.cache-dir "/root/.cache/pip"
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
echo "::endgroup::"
ldconfig
formers_test