fix tensorrt compiler bug

92118248 · Luo Tao · 19e877ff · 92118248
隐藏空白更改
内联并排

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)