Correct supported architectures for Cuda 11+

Kitware · Jan 16, 2023 · 464d439 · 464d439
1 parent 45ff09c
commit 464d439
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/CMake/kwiver-depends-CUDA.cmake b/CMake/kwiver-depends-CUDA.cmake
@@ -15,7 +15,7 @@ if( KWIVER_ENABLE_CUDA )
   if(CMAKE_VERSION VERSION_GREATER_EQUAL "3.18")
     cmake_policy(SET CMP0104 NEW)
     if(NOT DEFINED CMAKE_CUDA_ARCHITECTURES)
-      set(CMAKE_CUDA_ARCHITECTURES 35 50 52 60 61)
+      set(CMAKE_CUDA_ARCHITECTURES 52 60 61)
       if(CUDA_VERSION VERSION_GREATER_EQUAL "9.0")
         list(APPEND CMAKE_CUDA_ARCHITECTURES 70)
       endif()
@@ -25,7 +25,13 @@ if( KWIVER_ENABLE_CUDA )
       if(CUDA_VERSION VERSION_GREATER_EQUAL "11.0")
         list(APPEND CMAKE_CUDA_ARCHITECTURES 80)
       endif()
+
+      if(CUDA_VERSION VERSION_LESS_EQUAL "10.0")
+        list(APPEND CMAKE_CUDA_ARCHITECTURES 35 50)
+      endif()
     endif()
   endif()
   message(STATUS "CUDA Architectures: ${CMAKE_CUDA_ARCHITECTURES}")
+  message(STATUS "CUDA Version: ${CUDA_VERSION}")
+  message(STATUS "CUDA Compiler: ${CMAKE_CUDA_COMPILER}")
 endif( KWIVER_ENABLE_CUDA )