@@ -62,22 +62,20 @@ endif()
62
62
63
63
if (WITH_ANAKIN ) # only needed in CI
64
64
# compile the libinference_anakin_api.a and anakin.so.
65
- nv_library (inference_anakin_api SRCS api.cc api_anakin_engine.cc DEPS anakin )
66
- nv_library (inference_anakin_api_shared SHARED SRCS api.cc api_anakin_engine.cc DEPS anakin )
67
- set (ANAKIN_DEPS_LIB ${ANAKIN_SHARED_LIB} ${ANAKIN_SABER_LIB} ${CUDNN_LIBRARY} )
65
+ nv_library (inference_anakin_api SRCS api.cc api_anakin_engine.cc DEPS anakin_shared anakin_saber )
66
+ #nv_library(inference_anakin_api_shared SHARED SRCS api.cc api_anakin_engine.cc DEPS anakin)
68
67
function (anakin_target target_name )
69
68
target_compile_options (${target_name} BEFORE PUBLIC ${ANAKIN_COMPILE_EXTRA_FLAGS} )
70
- target_link_libraries (${target_name} ${ANAKIN_DEPS_LIB} )
71
69
endfunction ()
72
70
anakin_target (inference_anakin_api )
73
- anakin_target (inference_anakin_api_shared )
71
+ # anakin_target(inference_anakin_api_shared)
74
72
if (WITH_TESTING )
75
73
# TODO(luotao): anakin_model will move to demo_ci later
76
74
set (ANAKIN_MODLE_URL "http://paddle-inference-dist.bj.bcebos.com/mobilenet_v2.anakin.bin" )
77
- execute_process (COMMAND bash -c "cd ${ANAKIN_INSTALL_DIR} ; wget -q --no-check-certificate ${ANAKIN_MODLE_URL} " )
75
+ execute_process (COMMAND bash -c "wget -q --no-check-certificate ${ANAKIN_MODLE_URL} " )
78
76
cc_test (inference_anakin_test SRCS api_anakin_engine_tester.cc
79
- ARGS --model=${ANAKIN_INSTAL_DIR }/mobilenet_v2.anakin.bin
80
- DEPS inference_anakin_api_shared dynload_cuda SERIAL )
77
+ ARGS --model=${CMAKE_CURRENT_BINARY_DIR }/mobilenet_v2.anakin.bin
78
+ DEPS inference_anakin_api dynload_cuda SERIAL )
81
79
target_compile_options (inference_anakin_test BEFORE PUBLIC ${ANAKIN_COMPILE_EXTRA_FLAGS} )
82
80
endif (WITH_TESTING )
83
81
endif ()
0 commit comments