chore: set CUDA_ARCHITECTURE. This commit should not be merged.

Signed-off-by: Shin-kyoto <[email protected]>
autowarefoundation · May 30, 2024 · 327c358 · 327c358
1 parent 441a6ff
commit 327c358
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/perception/tensorrt_yolox/CMakeLists.txt b/perception/tensorrt_yolox/CMakeLists.txt
@@ -1,6 +1,9 @@
 cmake_minimum_required(VERSION 3.17)
 project(tensorrt_yolox)
 
+cmake_policy(SET CMP0104 NEW)
+set(CMAKE_CUDA_ARCHITECTURES 86)
+
 find_package(tensorrt_common)
 if(NOT ${tensorrt_common_FOUND})
   message(WARNING "The tensorrt_common package is not found. Please check its dependencies.")

diff --git a/perception/traffic_light_fine_detector/CMakeLists.txt b/perception/traffic_light_fine_detector/CMakeLists.txt
@@ -1,6 +1,9 @@
 cmake_minimum_required(VERSION 3.14)
 project(traffic_light_fine_detector)
 
+cmake_policy(SET CMP0104 NEW)
+set(CMAKE_CUDA_ARCHITECTURES 86)
+
 find_package(autoware_cmake REQUIRED)
 autoware_package()