fix tensorrt compiler bug

7 years ago · 9211824841
parent 19e877ffdb
commit 9211824841
1 changed files with 4 additions and 1 deletions
--- 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)