diff --git a/paddle/scripts/paddle_build.sh b/paddle/scripts/paddle_build.sh
index a05761ea6dc245886f3a159b9decae0f7578fbc0..b5c44f3d55b39bfc97434af30d1223ae1ff40d53 100755
--- a/paddle/scripts/paddle_build.sh
+++ b/paddle/scripts/paddle_build.sh
@@ -896,9 +896,9 @@ function card_test() {
     set -m
     case_count $1 $2
     ut_startTime_s=`date +%s` 
-    # get the CUDA device count
+    # get the CUDA device count, XPU device count is one
     if [ "${WITH_XPU}" == "ON" ];then
-        CUDA_DEVICE_COUNT=4
+        CUDA_DEVICE_COUNT=1
     else
         CUDA_DEVICE_COUNT=$(nvidia-smi -L | wc -l)
     fi