Merge pull request #10545 from luotao1/fix_tensorrt_engine

fix tensorrt_engine compiler error
7 years ago · 28de0ea404
parent 62c51e44d2 4a5ebb6806
commit 28de0ea404
2 changed files with 2 additions and 2 deletions
--- a/paddle/fluid/inference/CMakeLists.txt
+++ b/paddle/fluid/inference/CMakeLists.txt
@ -20,8 +20,8 @@ if(NOT APPLE)
 endif()

 if(WITH_TESTING)
+  # both tests/book and analysis depends the models that generated by python/paddle/fluid/tests/book
  add_subdirectory(tests/book)
-  # analysis test depends the models that generate by python/paddle/fluid/tests/book
  add_subdirectory(analysis)
 endif()

--- a/paddle/fluid/inference/tensorrt/CMakeLists.txt
+++ b/paddle/fluid/inference/tensorrt/CMakeLists.txt
@ -1,4 +1,4 @@
-nv_library(tensorrt_engine SRCS engine.cc)
+nv_library(tensorrt_engine SRCS engine.cc DEPS framework_proto)
 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)