From 26213a7732f319b7a377c25d7d179fdf3373224b Mon Sep 17 00:00:00 2001
From: tianshuo78520a <707759223@qq.com>
Date: Thu, 19 Aug 2021 09:47:31 +0800
Subject: [PATCH] Fix Inference CI CPU/GPU (#34931)

* notest;test=gpu-inference

* notest;test=gpu-inference

* notest;test=gpu-inference

* notest;test=gpu-inference

* fix error

* notest;test=gpu-inference

* notest;test=gpu-inference

* notest;test=gpu-inference

* test=gpu-inference
---
 paddle/fluid/inference/goapi/test.sh | 2 +-
 paddle/scripts/paddle_build.sh       | 5 +++++
 2 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/paddle/fluid/inference/goapi/test.sh b/paddle/fluid/inference/goapi/test.sh
index b764e2ac72c..ae4b8d97d6c 100644
--- a/paddle/fluid/inference/goapi/test.sh
+++ b/paddle/fluid/inference/goapi/test.sh
@@ -21,7 +21,7 @@ if [ ! -d mobilenetv1 ]; then
 fi
 
 # 2. set LD_LIBRARY_PATH
-export LD_LIBRARY_PATH=${LD_LIBRARY_PATH}:$PWD/paddle_inference_c/paddle/lib
+export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$PWD/paddle_inference_c/third_party/install/mklml/lib/:$PWD/paddle_inference_c/third_party/install/mkldnn/lib/:$PWD/paddle_inference_c/paddle/lib/
 
 # 3. go test
 go test -v ./...
diff --git a/paddle/scripts/paddle_build.sh b/paddle/scripts/paddle_build.sh
index abaae9a361d..0ace9568d49 100755
--- a/paddle/scripts/paddle_build.sh
+++ b/paddle/scripts/paddle_build.sh
@@ -2422,6 +2422,11 @@ function main() {
         python ${PADDLE_ROOT}/tools/remove_grad_op_and_kernel.py
         gen_fluid_lib ${parallel_number}
         ;;
+      gpu_inference)
+        test_fluid_lib
+        test_go_inference_api
+        check_approvals_of_unittest 3
+        ;;
       test_train)
         gen_fluid_lib ${parallel_number}
         test_fluid_lib_train
-- 
GitLab