Skip to content

Commit 44b6a5f

Browse files
committed
fix inference_lib.cmake and make demo_ci pass
1 parent af1e54a commit 44b6a5f

File tree

6 files changed

+22
-25
lines changed

6 files changed

+22
-25
lines changed

cmake/inference_lib.cmake

Lines changed: 14 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -138,25 +138,22 @@ copy(memory_lib
138138

139139
set(inference_deps paddle_fluid_shared paddle_fluid)
140140

141-
if(WITH_CONTRIB)
142-
message(STATUS "installing contrib")
143-
set(contrib_dst_dir "${FLUID_INSTALL_DIR}/contrib/inference")
144-
if (WITH_ANAKIN AND WITH_GPU)
145-
copy(contrib_anakin_inference_lib DEPS paddle_inference_api inference_anakin_api
146-
SRCS
147-
${PADDLE_BINARY_DIR}/paddle/contrib/inference/libinference_anakin_api* # compiled anakin api
148-
${PADDLE_BINARY_DIR}/third_party/install/anakin/*.tar.gz # anakin release
149-
DSTS ${contrib_dst_dir}/anakin ${contrib_dst_dir}/anakin)
150-
list(APPEND inference_deps contrib_anakin_inference_lib)
151-
endif()
152-
153-
copy(contrib_inference_lib DEPS paddle_inference_api paddle_inference_api_shared
154-
SRCS ${PADDLE_SOURCE_DIR}/paddle/contrib/inference/paddle_inference_api.h
155-
${PADDLE_BINARY_DIR}/paddle/contrib/inference/libpaddle_inference_api*
156-
DSTS ${contrib_dst_dir} ${contrib_dst_dir})
157-
list(APPEND inference_deps contrib_inference_lib)
141+
set(module "inference/api")
142+
if (WITH_ANAKIN AND WITH_GPU)
143+
copy(anakin_inference_lib DEPS paddle_inference_api inference_anakin_api
144+
SRCS
145+
${PADDLE_BINARY_DIR}/paddle/fluid/inference/api/libinference_anakin_api* # compiled anakin api
146+
${PADDLE_BINARY_DIR}/third_party/install/anakin/*.tar.gz # anakin release
147+
DSTS ${dst_dir}/inference/anakin ${dst_dir}/inference/anakin)
148+
list(APPEND inference_deps anakin_inference_lib)
158149
endif()
159150

151+
copy(inference_api_lib DEPS paddle_inference_api paddle_inference_api_shared
152+
SRCS ${src_dir}/${module}/api.h ${PADDLE_BINARY_DIR}/paddle/fluid/inference/api/libpaddle_inference_api*
153+
DSTS ${dst_dir}/inference ${dst_dir}/inference
154+
)
155+
list(APPEND inference_deps inference_api_lib)
156+
160157
set(module "inference")
161158
copy(inference_lib DEPS ${inference_deps}
162159
SRCS ${src_dir}/${module}/*.h ${PADDLE_BINARY_DIR}/paddle/fluid/inference/libpaddle_fluid.*

paddle/fluid/inference/api/demo_ci/CMakeLists.txt

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -55,11 +55,11 @@ endif()
5555
# Note: libpaddle_inference_api.so/a must put before libpaddle_fluid.so/a
5656
if(WITH_STATIC_LIB)
5757
set(DEPS
58-
${PADDLE_LIB}/contrib/inference/libpaddle_inference_api.a
58+
${PADDLE_LIB}/paddle/fluid/inference/libpaddle_inference_api.a
5959
${PADDLE_LIB}/paddle/fluid/inference/libpaddle_fluid.a)
6060
else()
6161
set(DEPS
62-
${PADDLE_LIB}/contrib/inference/libpaddle_inference_api.so
62+
${PADDLE_LIB}/paddle/fluid/inference/libpaddle_inference_api.so
6363
${PADDLE_LIB}/paddle/fluid/inference/libpaddle_fluid.so)
6464
endif()
6565
set(EXTERNAL_LIB "-lrt -ldl -lpthread")

paddle/fluid/inference/api/demo_ci/run.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -64,7 +64,7 @@ for WITH_STATIC_LIB in ON OFF; do
6464
-DWITH_GPU=$TEST_GPU_CPU \
6565
-DWITH_STATIC_LIB=$WITH_STATIC_LIB
6666
make -j
67-
for use_gpu in false; do
67+
for use_gpu in $use_gpu_list; do
6868
for vis_demo_name in $vis_demo_list; do
6969
./vis_demo \
7070
--modeldir=../data/$vis_demo_name/model \

paddle/fluid/inference/api/demo_ci/simple_on_word2vec.cc

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -19,8 +19,8 @@ limitations under the License. */
1919
#include <gflags/gflags.h>
2020
#include <glog/logging.h>
2121
#include <memory>
22-
#include <thread>
23-
#include "contrib/inference/paddle_inference_api.h"
22+
#include <thread> //NOLINT
23+
#include "paddle/fluid/inference/api.h"
2424
#include "paddle/fluid/platform/enforce.h"
2525

2626
DEFINE_string(dirname, "", "Directory of the inference model.");

paddle/fluid/inference/api/demo_ci/utils.h

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -13,10 +13,10 @@
1313
// limitations under the License.
1414

1515
#pragma once
16+
#include <algorithm>
1617
#include <string>
1718
#include <vector>
18-
19-
#include "contrib/inference/paddle_inference_api.h"
19+
#include "paddle/fluid/inference/api.h"
2020

2121
namespace paddle {
2222
namespace demo {

paddle/scripts/paddle_build.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -531,7 +531,7 @@ function test_fluid_inference_lib() {
531531
Testing fluid inference library ...
532532
========================================
533533
EOF
534-
cd ${PADDLE_ROOT}/paddle/contrib/inference/demo_ci
534+
cd ${PADDLE_ROOT}/paddle/fluid/inference/api/demo_ci
535535
./run.sh ${PADDLE_ROOT} ${WITH_MKL:-ON} ${WITH_GPU:-OFF}
536536
fi
537537
}

0 commit comments

Comments
 (0)