diff --git a/scripts/unittest_check_gpu.sh b/scripts/unittest_check_gpu.sh index ada1a49e4..b2d801d64 100644 --- a/scripts/unittest_check_gpu.sh +++ b/scripts/unittest_check_gpu.sh @@ -46,8 +46,8 @@ python -m pip install -r requirements.txt echo '************************************************************************************************************' echo "Checking code unit test by pytest ..." -python -m pip install pytest-timeout pytest-xdist pytest-rerunfailures -python -m pytest -n 1 --reruns=3 ./tests; check_error=$? +python -m pip install pytest-timeout pytest-xdist pytest-rerunfailures pytest-forked +python -m pytest --forked -n 1 --reruns=3 ./tests; check_error=$? if [ ${check_error} != 0 ];then echo "Rerun unit test check." python -m pytest -n 1 --lf ./tests; check_error=$?