From 3378a1302ab5a2feead4fadff2bdefcd31950881 Mon Sep 17 00:00:00 2001 From: zhangjun Date: Wed, 6 Jan 2021 12:34:24 +0800 Subject: [PATCH] update cmake --- cmake/paddlepaddle.cmake | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/cmake/paddlepaddle.cmake b/cmake/paddlepaddle.cmake index ca40c7b5..9ca68859 100644 --- a/cmake/paddlepaddle.cmake +++ b/cmake/paddlepaddle.cmake @@ -39,6 +39,12 @@ if (WITH_GPU) else() SET(PADDLE_LIB_VERSION "${PADDLE_VERSION}-gpu-cuda10-cudnn7-avx-mkl") endif() +elseif (WITH_LITE) + if (WITH_XPU) + SET(PADDLE_LIB_VERSION "${PADDLE_VERSION}-arm-xpu") + else() + SET(PADDLE_LIB_VERSION "${PADDLE_VERSION}-arm") + endif() else() if (WITH_AVX) if (WITH_MKLML) @@ -52,7 +58,7 @@ else() endif() if(WITH_LITE) - SET(PADDLE_LIB_PATH "/home/phytium/houjue/serving_tmp/build_paddle/fluid_inference.tgz") + SET(PADDLE_LIB_PATH "http://paddle-serving.bj.bcebos.com/inferlib/${PADDLE_LIB_VERSION}/fluid_inference.tgz") else() SET(PADDLE_LIB_PATH "http://paddle-inference-lib.bj.bcebos.com/${PADDLE_LIB_VERSION}/fluid_inference.tgz") endif() -- GitLab