Merge pull request #12811 from luotao1/tensorrt_compiler_bug

fix tensorrt compiler bug

Merge pull request #12811 from luotao1/tensorrt_compiler_bug
fix tensorrt compiler bug
2ae885e2 · Tao Luo · GitHub · 57b34d91 · 808e5b17 · 2ae885e2
显示空白变更内容
内联并排

Showing with 1 addition and 1 deletion

paddle/fluid/inference/tensorrt/CMakeLists.txt paddle/fluid/inference/tensorrt/CMakeLists.txt +1 -1

未找到文件。
--- a/paddle/fluid/inference/tensorrt/CMakeLists.txt
+++ b/paddle/fluid/inference/tensorrt/CMakeLists.txt
-nv_library(tensorrt_engine SRCS engine.cc DEPS framework_proto)
+nv_library(tensorrt_engine SRCS engine.cc DEPS framework_proto device_context)
 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(convert)