diff --git a/tools/serving_build.sh b/tools/serving_build.sh
index 3a2ea2f2b66effae1f5597c66196cb2dd0584c05..aec495581a6085168e72d4b3209db59e2d88630f 100644
--- a/tools/serving_build.sh
+++ b/tools/serving_build.sh
@@ -151,30 +151,41 @@ function python_run_criteo_ctr_with_cube() {
     local TYPE=$1
     yum install -y bc >/dev/null
     cd criteo_ctr_with_cube # pwd: /Serving/python/examples/criteo_ctr_with_cube
-    check_cmd "wget https://paddle-serving.bj.bcebos.com/unittest/ctr_cube_unittest.tar.gz"
-    check_cmd "tar xf ctr_cube_unittest.tar.gz"
-    check_cmd "mv models/ctr_client_conf ./"
-    check_cmd "mv models/ctr_serving_model_kv ./"
-    check_cmd "mv models/data ./cube/"
-    check_cmd "mv models/ut_data ./"
-    cp ../../../build-server-$TYPE/output/bin/cube* ./cube/ 
-    mkdir -p $PYTHONROOT/lib/python2.7/site-packages/paddle_serving_server/serving-cpu-avx-openblas-0.1.3/
-    yes | cp ../../../build-server-$TYPE/output/demo/serving/bin/serving $PYTHONROOT/lib/python2.7/site-packages/paddle_serving_server/serving-cpu-avx-openblas-0.1.3/
+    case $TYPE in
+        CPU)
+            check_cmd "wget https://paddle-serving.bj.bcebos.com/unittest/ctr_cube_unittest.tar.gz"
+            check_cmd "tar xf ctr_cube_unittest.tar.gz"
+            check_cmd "mv models/ctr_client_conf ./"
+            check_cmd "mv models/ctr_serving_model_kv ./"
+            check_cmd "mv models/data ./cube/"
+            check_cmd "mv models/ut_data ./"
+            cp ../../../build-server-$TYPE/output/bin/cube* ./cube/ 
+            mkdir -p $PYTHONROOT/lib/python2.7/site-packages/paddle_serving_server/serving-cpu-avx-openblas-0.1.3/
+            yes | cp ../../../build-server-$TYPE/output/demo/serving/bin/serving $PYTHONROOT/lib/python2.7/site-packages/paddle_serving_server/serving-cpu-avx-openblas-0.1.3/
 
-    sh cube_prepare.sh &
-    check_cmd "mkdir work_dir1 && cp cube/conf/cube.conf ./work_dir1/"    
-    python test_server.py ctr_serving_model_kv &
-    check_cmd "python test_client.py ctr_client_conf/serving_client_conf.prototxt ./ut_data >score"
-    AUC=$(tail -n 2  score | awk 'NR==1')
-    VAR2="0.70"
-    RES=$( echo "$AUC>$VAR2" | bc )
-    if [[ $RES -eq 0 ]]; then
-        echo "error with criteo_ctr_with_cube inference auc test, auc should > 0.70"
-        exit 1
-    fi
-    echo "criteo_ctr_with_cube inference auc test success"
-    ps -ef | grep "paddle_serving_server" | grep -v grep | awk '{print $2}' | xargs kill
-    ps -ef | grep "cube" | grep -v grep | awk '{print $2}' | xargs kill
+            sh cube_prepare.sh &
+            check_cmd "mkdir work_dir1 && cp cube/conf/cube.conf ./work_dir1/"    
+            python test_server.py ctr_serving_model_kv &
+            check_cmd "python test_client.py ctr_client_conf/serving_client_conf.prototxt ./ut_data >score"
+            AUC=$(tail -n 2  score | awk 'NR==1')
+            VAR2="0.70"
+            RES=$( echo "$AUC>$VAR2" | bc )
+            if [[ $RES -eq 0 ]]; then
+                echo "error with criteo_ctr_with_cube inference auc test, auc should > 0.70"
+                exit 1
+            fi
+            echo "criteo_ctr_with_cube inference auc test success"
+            ps -ef | grep "paddle_serving_server" | grep -v grep | awk '{print $2}' | xargs kill
+            ps -ef | grep "cube" | grep -v grep | awk '{print $2}' | xargs kill
+            ;;
+        GPU)
+            ;;
+        *)
+            echo "error type"
+            exit 1
+            ;;
+    esac
+    echo "test criteo_ctr_with_cube $TYPE part finished as expected."
     cd .. # pwd: /Serving/python/examples
 }