File tree Expand file tree Collapse file tree 2 files changed +6
-4
lines changed
paddle/fluid/inference/api Expand file tree Collapse file tree 2 files changed +6
-4
lines changed Original file line number Diff line number Diff line change @@ -11,6 +11,11 @@ set(ANAKIN_LIBRARY ${ANAKIN_INSTALL_DIR})
11
11
set (ANAKIN_SHARED_LIB ${ANAKIN_LIBRARY} /libanakin.so )
12
12
set (ANAKIN_SABER_LIB ${ANAKIN_LIBRARY} /libanakin_saber_common.so )
13
13
14
+ # TODO(luotao): ANAKIN_MODLE_URL will move to demo ci later.
15
+ set (ANAKIN_MODLE_URL "http://paddle-inference-dist.bj.bcebos.com/mobilenet_v2.anakin.bin" )
16
+ execute_process (COMMAND bash -c "mkdir -p ${ANAKIN_SOURCE_DIR} " )
17
+ execute_process (COMMAND bash -c "cd ${ANAKIN_SOURCE_DIR} ; wget -q --no-check-certificate ${ANAKIN_MODLE_URL} " )
18
+
14
19
include_directories (${ANAKIN_INCLUDE} )
15
20
include_directories (${ANAKIN_INCLUDE} /saber/ )
16
21
Original file line number Diff line number Diff line change @@ -70,11 +70,8 @@ if (WITH_ANAKIN) # only needed in CI
70
70
anakin_target (inference_anakin_api )
71
71
#anakin_target(inference_anakin_api_shared)
72
72
if (WITH_TESTING )
73
- # TODO(luotao): anakin_model will move to demo_ci later
74
- set (ANAKIN_MODLE_URL "http://paddle-inference-dist.bj.bcebos.com/mobilenet_v2.anakin.bin" )
75
- execute_process (COMMAND bash -c "wget -q --no-check-certificate ${ANAKIN_MODLE_URL} " )
76
73
cc_test (inference_anakin_test SRCS api_anakin_engine_tester.cc
77
- ARGS --model=${CMAKE_CURRENT_BINARY_DIR }/mobilenet_v2.anakin.bin
74
+ ARGS --model=${ANAKIN_SOURCE_DIR }/mobilenet_v2.anakin.bin
78
75
DEPS inference_anakin_api dynload_cuda SERIAL )
79
76
target_compile_options (inference_anakin_test BEFORE PUBLIC ${ANAKIN_COMPILE_EXTRA_FLAGS} )
80
77
endif (WITH_TESTING )
You can’t perform that action at this time.
0 commit comments