Merge pull request #10545 from luotao1/fix_tensorrt_engine

luotao1 · web-flow · commit 28de0ea404f6 · 2018-05-10T12:43:20.000+08:00
fix tensorrt_engine compiler error
diff --git 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()
 
diff --git 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)
 

Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,4 @@`
`1`		`-nv_library(tensorrt_engine SRCS engine.cc)`
	`1`	`+nv_library(tensorrt_engine SRCS engine.cc DEPS framework_proto)`
`2`	`2`	`nv_test(test_tensorrt SRCS test_tensorrt.cc DEPS dynload_cuda device_context dynamic_loader)`
`3`	`3`	`nv_test(test_tensorrt_engine SRCS test_engine.cc DEPS dynload_cuda tensorrt_engine)`
`4`	`4`