From 86a33985470b4bf12e43a651d7660bd588024fd2 Mon Sep 17 00:00:00 2001
From: chang zherui <760161589@qq.com>
Date: Fri, 3 Apr 2020 09:38:04 +0800
Subject: [PATCH] modify runtest py ut for parallel exec

---
 tests/ut/python/runtest.sh | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/tests/ut/python/runtest.sh b/tests/ut/python/runtest.sh
index d1c3299821..035e454f24 100755
--- a/tests/ut/python/runtest.sh
+++ b/tests/ut/python/runtest.sh
@@ -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=$?