File tree Expand file tree Collapse file tree 3 files changed +5
-5
lines changed
paddle/fluid/inference/api Expand file tree Collapse file tree 3 files changed +5
-5
lines changed Original file line number Diff line number Diff line change @@ -204,11 +204,12 @@ include(external/snappy) # download snappy
204
204
include (external/snappystream )
205
205
include (external/threadpool )
206
206
207
- set (WITH_ANAKIN OFF CACHE STRING "Disable Anakin first, will add it later." FORCE )
208
207
if (WITH_GPU )
209
208
include (cuda )
210
209
include (tensorrt )
211
210
include (external/anakin )
211
+ elseif ()
212
+ set (WITH_ANAKIN OFF CACHE STRING "Anakin is used in GPU only now." FORCE )
212
213
endif ()
213
214
214
215
include (cudnn ) # set cudnn libraries, must before configure
Original file line number Diff line number Diff line change @@ -35,9 +35,8 @@ set(ANAKIN_COMPILE_EXTRA_FLAGS
35
35
ExternalProject_Add (
36
36
extern_anakin
37
37
${EXTERNAL_PROJECT_LOG_ARGS}
38
- # TODO(luotao): use PaddlePaddle/Anakin later
39
- GIT_REPOSITORY "https://github.com/luotao1/Anakin"
40
- GIT_TAG "842a89ae3747ede25d8acbc29030d2eb602ced1f"
38
+ GIT_REPOSITORY "https://github.com/PaddlePaddle/Anakin"
39
+ GIT_TAG "04256ba78fa3da0beb74e8036c8efd68c12824d6"
41
40
PREFIX ${ANAKIN_SOURCE_DIR}
42
41
UPDATE_COMMAND ""
43
42
CMAKE_ARGS -DUSE_GPU_PLACE=YES
Original file line number Diff line number Diff line change @@ -60,7 +60,7 @@ cc_library(paddle_inference_tensorrt_subgraph_engine
60
60
inference_api_test (test_api_tensorrt_subgraph_engine SRC api_tensorrt_subgraph_engine_tester.cc ARGS test_word2vec )
61
61
endif ()
62
62
63
- if (WITH_ANAKIN ) # only needed in CI
63
+ if (WITH_ANAKIN AND WITH_GPU ) # only needed in CI
64
64
# compile the libinference_anakin_api.a and anakin.so.
65
65
nv_library (inference_anakin_api SRCS api.cc api_anakin_engine.cc DEPS anakin_shared anakin_saber )
66
66
#nv_library(inference_anakin_api_shared SHARED SRCS api.cc api_anakin_engine.cc DEPS anakin)
You can’t perform that action at this time.
0 commit comments