Fix cmake find trtexec bin

autowarefoundation · Nov 1, 2023 · bf7f263 · bf7f263
1 parent 2ee44a3
commit bf7f263
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/common/trtexec_vendor/CMakeLists.txt b/common/trtexec_vendor/CMakeLists.txt
@@ -28,7 +28,8 @@ if(TENSORRT_VERSION VERSION_LESS 8.2.1)
 endif()
 
 set(TRTEXEC_DEFAULT_BIN /usr/src/tensorrt/bin/trtexec)
-if(NOT EXISTS TRTEXEC_DEFAULT_BIN)
+if(NOT EXISTS ${TRTEXEC_DEFAULT_BIN})
+  message(STATUS "${TRTEXEC_DEFAULT_BIN} not found, download and build from source...")
   include(FetchContent)
   fetchcontent_declare(tensorrt
     GIT_REPOSITORY https://github.com/NVIDIA/TensorRT
@@ -71,6 +72,8 @@ if(NOT EXISTS TRTEXEC_DEFAULT_BIN)
     LIBRARY DESTINATION lib
     RUNTIME DESTINATION bin
   )
+else()  
+  message(STATUS "Found trtexec: ${TRTEXEC_DEFAULT_BIN}")
 endif()
 
 if(BUILD_TESTING)