From 82740b1f3c28979a9ea613b71ce7e60050f6252c Mon Sep 17 00:00:00 2001
From: tianshuo78520a <707759223@qq.com>
Date: Mon, 28 Sep 2020 11:49:02 +0800
Subject: [PATCH] notest;test=kunlun

---
 paddle/scripts/paddle_build.sh                     | 9 +++++++--
 python/paddle/fluid/tests/unittests/CMakeLists.txt | 4 ++++
 2 files changed, 11 insertions(+), 2 deletions(-)

diff --git a/paddle/scripts/paddle_build.sh b/paddle/scripts/paddle_build.sh
index c71ff47e850..455741c7026 100755
--- a/paddle/scripts/paddle_build.sh
+++ b/paddle/scripts/paddle_build.sh
@@ -895,7 +895,11 @@ function card_test() {
     case_count $1 $2
     ut_startTime_s=`date +%s` 
     # get the CUDA device count
-    CUDA_DEVICE_COUNT=$(nvidia-smi -L | wc -l)
+    if [ "${WITH_XPU}" == "ON" ];then
+        CUDA_DEVICE_COUNT=$(xpu_smi -m 2 | wc -l)
+    else
+        CUDA_DEVICE_COUNT=$(nvidia-smi -L | wc -l)
+    fi
 
     testcases=$1
     if (( $# > 1 )); then
@@ -1173,7 +1177,8 @@ function parallel_test_base_xpu() {
     ========================================
 EOF
         ut_startTime_s=`date +%s`
-        ctest -R *xpu 
+        exclusive_tests=$(ctest -N |grep "_xpu")        # cases list which would be run exclusively
+        card_test $exclusive_tests 1
         ut_endTime_s=`date +%s`
         echo "XPU testCase Time: $[ $ut_endTime_s - $ut_startTime_s ]s"
         if [[ "$EXIT_CODE" != "0" ]]; then
diff --git a/python/paddle/fluid/tests/unittests/CMakeLists.txt b/python/paddle/fluid/tests/unittests/CMakeLists.txt
index 97a3ebc2135..0b06958b916 100644
--- a/python/paddle/fluid/tests/unittests/CMakeLists.txt
+++ b/python/paddle/fluid/tests/unittests/CMakeLists.txt
@@ -546,6 +546,10 @@ add_subdirectory(sequence)
 add_subdirectory(dygraph_to_static)
 add_subdirectory(rnn)
 
+if (WITH_XPU)
+    add_subdirectory(xpu)
+endif()
+
 if (WITH_MKLDNN)
     add_subdirectory(mkldnn)
 endif()
-- 
GitLab