Merge pull request #11794 from luotao1/fix_tensorrt_compiler_bug

fix tensorrt compiler bug

Merge pull request #11794 from luotao1/fix_tensorrt_compiler_bug
fix tensorrt compiler bug
905454df · Tao Luo · GitHub · 2ecc5622 · 92118248 · 905454df
隐藏空白更改
内联并排

Showing with 4 addition and 1 deletion

paddle/contrib/inference/CMakeLists.txt paddle/contrib/inference/CMakeLists.txt +4 -1

未找到文件。
--- a/paddle/contrib/inference/CMakeLists.txt
+++ b/paddle/contrib/inference/CMakeLists.txt
@@ -18,7 +18,10 @@ if(APPLE)
 endif(APPLE)


-set(inference_deps paddle_inference_api paddle_fluid_api paddle_inference_tensorrt_subgraph_engine)
+set(inference_deps paddle_inference_api paddle_fluid_api)
+if(WITH_GPU AND TENSORRT_FOUND)
+    set(inference_deps ${inference_deps} paddle_inference_tensorrt_subgraph_engine)
+endif()

 function(inference_api_test TARGET_NAME)
    if (WITH_TESTING)