From bc6b0ca1f44a5444b421c8219ede2a4cf4a7b2f4 Mon Sep 17 00:00:00 2001 From: nhzlx Date: Thu, 18 Apr 2019 11:07:18 +0000 Subject: [PATCH] fix trt anakin subgraph compile rely test=develop --- paddle/fluid/inference/anakin/CMakeLists.txt | 4 ++-- paddle/fluid/inference/tensorrt/CMakeLists.txt | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/paddle/fluid/inference/anakin/CMakeLists.txt b/paddle/fluid/inference/anakin/CMakeLists.txt index e8fb56590..9ffe70471 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 f4977d08c..d82b88a77 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) -- GitLab