summaryrefslogtreecommitdiff
blob: c1cf2981e2ea05be843f8eddcf736014e72f4145 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -590,8 +590,14 @@ if(CUDA_FOUND)
         if (NOT DEFINED OSD_CUDA_NVCC_FLAGS)
             if (CUDA_VERSION_MAJOR LESS 6)
                 set( OSD_CUDA_NVCC_FLAGS --gpu-architecture compute_11 )
-            else()
+            elseif (CUDA_VERSION_MAJOR LESS 9)
                 set( OSD_CUDA_NVCC_FLAGS --gpu-architecture compute_20 )
+            elseif (CUDA_VERSION_MAJOR LESS 11)
+                set( OSD_CUDA_NVCC_FLAGS --gpu-architecture compute_30 )
+            elseif (CUDA_VERSION_MAJOR LESS 12)
+                set( OSD_CUDA_NVCC_FLAGS --gpu-architecture compute_35 )
+            else()
+                set( OSD_CUDA_NVCC_FLAGS --gpu-architecture compute_50 )
             endif()
         endif()
     endif()