Skip to content

Commit 88acf53

Browse files
committed
Add support for CUDA versions through 12.1.
1 parent c0dc3d8 commit 88acf53

File tree

1 file changed

+16
-4
lines changed

1 file changed

+16
-4
lines changed

cmake/ChooseCudaCC.cmake

Lines changed: 16 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -65,7 +65,7 @@ function(chooseCudaCC SUPPORTED_CC SUPPORTED_GENCODE_FLAGS)
6565

6666
set(CC_LIST_BY_SYSTEM_PROCESSOR "")
6767
if(CMAKE_SYSTEM_PROCESSOR IN_LIST OTHER_SUPPORTED_PROCESSORS)
68-
list(APPEND CC_LIST_BY_SYSTEM_PROCESSOR "20;21;30;35;50;52;60;61;70;75;80;86")
68+
list(APPEND CC_LIST_BY_SYSTEM_PROCESSOR "20;21;30;35;50;52;60;61;70;75;80;86;87;89;90")
6969
endif()
7070
if(CMAKE_SYSTEM_PROCESSOR IN_LIST TEGRA_SUPPORTED_PROCESSORS)
7171
list(APPEND CC_LIST_BY_SYSTEM_PROCESSOR "32;53;62;72")
@@ -78,10 +78,20 @@ function(chooseCudaCC SUPPORTED_CC SUPPORTED_GENCODE_FLAGS)
7878
# Default setting of the CUDA CC versions to compile.
7979
# Shortening the lists saves a lot of compile time.
8080
#
81-
set(CUDA_MIN_CC 20)
82-
set(CUDA_MAX_CC 86)
83-
if(CUDA_VERSION VERSION_GREATER_EQUAL 11.1)
81+
82+
# The current version last time this list was updated was CUDA 12.1.
83+
if(CUDA_VERSION VERSION_GREATER_EQUAL 12)
84+
set(CUDA_MIN_CC 50)
85+
set(CUDA_MAX_CC 90)
86+
elseif(CUDA_VERSION VERSION_GREATER_EQUAL 11.8)
87+
set(CUDA_MIN_CC 35)
88+
set(CUDA_MAX_CC 90)
89+
elseif(CUDA_VERSION VERSION_GREATER_EQUAL 11.6)
90+
set(CUDA_MIN_CC 35)
91+
set(CUDA_MAX_CC 87)
92+
elseif(CUDA_VERSION VERSION_GREATER_EQUAL 11.1)
8493
set(CUDA_MIN_CC 35)
94+
set(CUDA_MAX_CC 86)
8595
elseif(CUDA_VERSION_MAJOR GREATER_EQUAL 11)
8696
set(CUDA_MIN_CC 35)
8797
set(CUDA_MAX_CC 80)
@@ -92,8 +102,10 @@ function(chooseCudaCC SUPPORTED_CC SUPPORTED_GENCODE_FLAGS)
92102
set(CUDA_MIN_CC 30)
93103
set(CUDA_MAX_CC 72)
94104
elseif(CUDA_VERSION_MAJOR GREATER_EQUAL 8)
105+
set(CUDA_MIN_CC 20)
95106
set(CUDA_MAX_CC 62)
96107
elseif(CUDA_VERSION_MAJOR GREATER_EQUAL 7)
108+
set(CUDA_MIN_CC 20)
97109
set(CUDA_MAX_CC 53)
98110
else()
99111
message(FATAL_ERROR "We do not support a CUDA SDK below version 7.0")

0 commit comments

Comments
 (0)