"fix dso" (#13407)

7 years ago · bf1e76e401
parent 40c54db301
commit bf1e76e401
1 changed files with 2 additions and 0 deletions
--- a/cmake/tensorrt.cmake
+++ b/cmake/tensorrt.cmake
@ -16,7 +16,9 @@ find_library(TENSORRT_LIBRARY NAMES libnvinfer.so libnvinfer.a
    DOC "Path to TensorRT library.")

 if(TENSORRT_INCLUDE_DIR AND TENSORRT_LIBRARY)
+  if(WITH_DSO)
    set(TENSORRT_FOUND ON)
+  endif(WITH DSO)
 else()
    set(TENSORRT_FOUND OFF)
 endif()