fix trt anakin subgraph compile rely

test=develop
6 years ago · bc6b0ca1f4
parent cbdb8a17b1
commit bc6b0ca1f4
2 changed files with 4 additions and 4 deletions
--- 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)
--- 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)