Merge pull request #851 from bartoszek/develop

Fix build against cuda<=11
alicevision · Aug 10, 2020 · 32e626e · 32e626e
2 parents 5c04121 + 2464e87
commit 32e626e
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/src/CMakeLists.txt b/src/CMakeLists.txt
@@ -730,9 +730,13 @@ if(ALICEVISION_HAVE_CUDA)
 
   set(CUDA_NVCC_FLAGS_DEBUG   "${CUDA_NVCC_FLAGS_DEBUG};-G;-g")
   # set(CUDA_NVCC_FLAGS_RELEASE "${CUDA_NVCC_FLAGS_RELEASE};-O3")
-  if(NOT CUDA_VERSION_MAJOR LESS 9)
-    set(ALICEVISION_CUDA_CC_LIST_BASIC 30 35 50 52 60 61 62)
-  elseif(CUDA_VERSION_MAJOR GREATER 7)
+  if(CUDA_VERSION_MAJOR GREATER_EQUAL 11)
+    set(ALICEVISION_CUDA_CC_LIST_BASIC 35 50 52 60 61 62 70 72 75 80)
+  elseif(CUDA_VERSION_MAJOR GREATER_EQUAL 10)
+    set(ALICEVISION_CUDA_CC_LIST_BASIC 30 35 50 52 60 61 62 70 72 75)
+  elseif(CUDA_VERSION_MAJOR GREATER_EQUAL 9)
+    set(ALICEVISION_CUDA_CC_LIST_BASIC 30 35 50 52 60 61 62 70)
+  elseif(CUDA_VERSION_MAJOR GREATER_EQUAL 8)
     set(ALICEVISION_CUDA_CC_LIST_BASIC 20 30 35 50 52 60 61 62)
   else()
     set(ALICEVISION_CUDA_CC_LIST_BASIC 20 30 35 50 52 )