Merge pull request #11794 from luotao1/fix_tensorrt_compiler_bug

fix tensorrt compiler bug
7 years ago · 905454dfec
parent 2ecc56226d 9211824841
commit 905454dfec
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)