diff --git a/paddle/fluid/inference/anakin/CMakeLists.txt b/paddle/fluid/inference/anakin/CMakeLists.txt index e8fb56590563f49f920bfe71d160ec822cb3ca30..9ffe70471c42ca95953ebce10c9246c777d7b2a2 100644 --- a/paddle/fluid/inference/anakin/CMakeLists.txt +++ b/paddle/fluid/inference/anakin/CMakeLists.txt @@ -1,5 +1,5 @@ -cc_library(anakin_engine SRCS engine.cc DEPS framework_proto) -cc_library(anakin_op_teller SRCS op_teller.cc DEPS framework_proto) +cc_library(anakin_engine SRCS engine.cc DEPS framework_proto boost) +cc_library(anakin_op_teller SRCS op_teller.cc DEPS framework_proto boost) target_link_libraries(anakin_engine anakin anakin_saber_common) cc_test(test_anakin_engine SRCS test_anakin_engine.cc DEPS anakin_engine) add_subdirectory(convert) diff --git a/paddle/fluid/inference/tensorrt/CMakeLists.txt b/paddle/fluid/inference/tensorrt/CMakeLists.txt index f4977d08c4d051b8a528e122c47948c3c81d153c..d82b88a77a9898a24090614a6db3439fd1acd74f 100644 --- a/paddle/fluid/inference/tensorrt/CMakeLists.txt +++ b/paddle/fluid/inference/tensorrt/CMakeLists.txt @@ -1,5 +1,5 @@ -nv_library(tensorrt_engine SRCS engine.cc trt_int8_calibrator.cc DEPS ${GLOB_OPERATOR_DEPS} framework_proto device_context) -nv_library(tensorrt_op_teller SRCS op_teller.cc DEPS framework_proto) +nv_library(tensorrt_engine SRCS engine.cc trt_int8_calibrator.cc DEPS ${GLOB_OPERATOR_DEPS} framework_proto device_context boost) +nv_library(tensorrt_op_teller SRCS op_teller.cc DEPS framework_proto boost) nv_test(test_tensorrt SRCS test_tensorrt.cc DEPS dynload_cuda device_context dynamic_loader) nv_test(test_tensorrt_engine SRCS test_engine.cc DEPS dynload_cuda tensorrt_engine) add_subdirectory(plugin)