modify runtest py ut for parallel exec

This commit is contained in:
chang zherui 2020-04-03 09:38:04 +08:00
parent 8d554d4322
commit 9b9a85ba34
1 changed files with 6 additions and 2 deletions

View File

@ -39,9 +39,13 @@ if [ "x${ENABLE_GE}" == "xON" -o "x${ENABLE_GE}" == "xOn" -o "x${ENABLE_GE}" ==
fi
if [ $# -gt 0 ]; then
pytest -s --ignore=$1/pynative_mode $IGNORE_EXEC $1
pytest -s --ignore=$1/pynative_mode --ignore=$1/parallel --ignore=$1/train $IGNORE_EXEC $1
pytest -n 4 --dist=loadfile -v $1/parallel
pytest -n 4 --dist=loadfile -v $1/train
else
pytest --ignore=$CURRPATH/pynative_mode $IGNORE_EXEC $CURRPATH
pytest --ignore=$CURRPATH/pynative_mode --ignore=$CURRPATH/parallel --ignore=$CURRPATH/train $IGNORE_EXEC $CURRPATH
pytest -n 4 --dist=loadfile -v $CURRPATH/parallel
pytest -n 4 --dist=loadfile -v $CURRPATH/train
fi
RET=$?