diff --git a/tools/test_ci_model_benchmark.sh b/tools/test_ci_model_benchmark.sh
index 71498126165b451e1cca8fe579a07fd2cbc8bee4..6ed4041363dfb8a780a900e7f1d2aefd4b64f26b 100644
--- a/tools/test_ci_model_benchmark.sh
+++ b/tools/test_ci_model_benchmark.sh
@@ -16,7 +16,7 @@
 
 
 function check_whl {
-    bash -x paddle/scripts/paddle_build.sh build
+    bash -x paddle/scripts/paddle_build.sh build_only
     [ $? -ne 0 ] && echo "build paddle failed." && exit 1
     pip uninstall -y paddlepaddle_gpu
     pip install build/python/dist/*.whl
@@ -35,13 +35,14 @@ function check_whl {
         rm -rf ${PADDLE_ROOT}/build/third_party
     fi
 
-    bash -x paddle/scripts/paddle_build.sh build
+    bash -x paddle/scripts/paddle_build.sh build_only
     [ $? -ne 0 ] && echo "build paddle failed." && exit 1
-    unzip -q python/dist/*.whl -d /tmp/develop
+    unzip -q build/python/dist/*.whl -d /tmp/develop
 
     sed -i '/version.py/d' /tmp/pr/*/RECORD
     sed -i '/version.py/d' /tmp/develop/*/RECORD
     diff_whl=`diff /tmp/pr/*/RECORD /tmp/develop/*/RECORD|wc -l`
+    [ $? -ne 0 ] && echo "diff paddle whl failed." && exit 1
     if [ ${diff_whl} -eq 0 ];then
         echo "paddle whl does not diff in PR-CI-Model-benchmark, so skip this ci"
         echo "ipipe_log_param_isSkipTest_model_benchmark: 1"