fix: skip building tensorrt yolox packages if cuda is not found (auto…

…warefoundation#2412) * fix: skip building tensorrt yolox packages if cuda is not found Signed-off-by: Daisuke Nishimatsu <border_goldenmarket@yahoo.co.jp> * fix typo Signed-off-by: Daisuke Nishimatsu <border_goldenmarket@yahoo.co.jp>
tier4 · Dec 8, 2022 · ee3321e · ee3321e
1 parent d647678
commit ee3321e
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 4 deletions.
diff --git a/common/cuda_utils/CMakeLists.txt b/common/cuda_utils/CMakeLists.txt
@@ -4,7 +4,12 @@ project(cuda_utils)
 find_package(autoware_cmake REQUIRED)
 autoware_package()
 
-find_package(CUDA REQUIRED)
+find_package(CUDA)
+
+if(NOT ${CUDA_FOUND})
+  message(WARNING "cuda is not found, so the cuda_utils package won't be built.")
+  return()
+endif()
 
 install(
   DIRECTORY include/${PROJECT_NAME}/

diff --git a/common/tensorrt_common/CMakeLists.txt b/common/tensorrt_common/CMakeLists.txt
@@ -4,9 +4,9 @@ project(tensorrt_common)
 find_package(autoware_cmake REQUIRED)
 autoware_package()
 
-find_package(CUDA REQUIRED)
-find_package(CUDNN REQUIRED)
-find_package(TENSORRT REQUIRED)
+find_package(CUDA)
+find_package(CUDNN)
+find_package(TENSORRT)
 
 if(NOT (${CUDA_FOUND} AND ${CUDNN_FOUND} AND ${TENSORRT_FOUND}))
   message(WARNING "cuda, cudnn, tensorrt libraries are not found")

diff --git a/perception/tensorrt_yolox/CMakeLists.txt b/perception/tensorrt_yolox/CMakeLists.txt
@@ -1,6 +1,12 @@
 cmake_minimum_required(VERSION 3.5)
 project(tensorrt_yolox)
 
+find_package(tensorrt_common)
+if(NOT ${tensorrt_common_FOUND})
+  message(WARNING "The tensorrt_common package is not found. Please check its dependencies.")
+  return()
+endif()
+
 find_package(autoware_cmake REQUIRED)
 autoware_package()