diff --git a/paddle/fluid/inference/goapi/test.sh b/paddle/fluid/inference/goapi/test.sh index b764e2ac72c70e7689af6828c69d0a7bcb716d5b..ae4b8d97d6cbdae5547d31a8fa5cee50e0f86574 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 abaae9a361d6bb1ef311308a7a0510b004ede607..0ace9568d496526b1e1ee7d1547746ef8f9c909c 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