diff --git a/paddle/scripts/paddle_build.sh b/paddle/scripts/paddle_build.sh index 69303013d2a41a049276c0d1b03b9d902b555d23..a05761ea6dc245886f3a159b9decae0f7578fbc0 100755 --- a/paddle/scripts/paddle_build.sh +++ b/paddle/scripts/paddle_build.sh @@ -246,6 +246,7 @@ function cmake_base() { -DWITH_GRPC=${grpc_flag} -DWITH_GLOO=${gloo_flag} -DWITH_LITE=${WITH_LITE:-OFF} + -DWITH_XPU=${WITH_XPU:-OFF} -DLITE_GIT_TAG=develop ======================================== EOF @@ -277,6 +278,7 @@ EOF -DWITH_GRPC=${grpc_flag} \ -DWITH_GLOO=${gloo_flag} \ -DLITE_GIT_TAG=develop \ + -DWITH_XPU=${WITH_XPU:-OFF} \ -DWITH_LITE=${WITH_LITE:-OFF};build_error=$? if [ "$build_error" != 0 ];then exit 7; @@ -895,7 +897,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=4 + else + CUDA_DEVICE_COUNT=$(nvidia-smi -L | wc -l) + fi testcases=$1 if (( $# > 1 )); then @@ -1163,6 +1169,41 @@ EOF fi } +function parallel_test_base_xpu() { + mkdir -p ${PADDLE_ROOT}/build + cd ${PADDLE_ROOT}/build + if [ ${WITH_TESTING:-ON} == "ON" ] ; then + cat <