Skip to content

Commit 950ac99

Browse files
committed
Fix model unittest CI
1 parent d4c7bce commit 950ac99

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

.github/workflows/model-unittest-gpu.yml

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -116,10 +116,6 @@ jobs:
116116
cp -r $work_dir/../../../models ./models
117117
echo "Check models:"
118118
ls -l ./models
119-
echo "Check Cuda Version"
120-
python -c "import paddle; print(paddle.version.cuda()); print(paddle.version.cudnn()); print(paddle.is_compiled_with_cuda())"
121-
nvcc -V
122-
cat /usr/local/cuda/version.txt
123119
echo "Test Start"
124120
timeout 30m bash scripts/regression/ci_model_unittest.sh ${paddle_whl}
125121
'

scripts/regression/ci_model_unittest.sh

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -104,6 +104,10 @@ if [[ ${FLAGS_enable_CI} == "true" ]] || [[ ${FLAGS_enable_CE} == "true" ]];then
104104
echo ' Testing all model unittest cases '
105105
unset http_proxy && unset https_proxy
106106
set +e
107+
echo "Check Cuda Version"
108+
python -c "import paddle; print(paddle.version.cuda()); print(paddle.version.cudnn()); print(paddle.is_compiled_with_cuda())"
109+
nvcc -V
110+
cat /usr/local/cuda/version.txt
107111
PYTHONPATH=$(pwd) \
108112
COVERAGE_SOURCE=paddleformers \
109113
python -m pytest -s -v ${model_unittest_path} > ${log_path}/model_unittest.log 2>&1

0 commit comments

Comments
 (0)