Skip to content

Commit 722dba7

Browse files
Simplify
1 parent a5e763b commit 722dba7

File tree

1 file changed

+7
-22
lines changed

1 file changed

+7
-22
lines changed

CMakeLists.txt

Lines changed: 7 additions & 22 deletions
Original file line numberDiff line numberDiff line change
@@ -93,8 +93,8 @@ if(BUILD_CUDA)
9393
)
9494
endif()
9595

96-
if(CMAKE_CUDA_COMPILER_VERSION VERSION_LESS "11.0")
97-
message(FATAL_ERROR "CUDA Version < 11 is not supported")
96+
if(CMAKE_CUDA_COMPILER_VERSION VERSION_LESS "11.4")
97+
message(FATAL_ERROR "CUDA Version < 11.4 is not supported")
9898
elseif(CMAKE_CUDA_COMPILER_VERSION VERSION_GREATER_EQUAL "13.0")
9999
message(FATAL_ERROR "CUDA Version > 12 is not supported")
100100
endif()
@@ -103,35 +103,20 @@ if(BUILD_CUDA)
103103
if(CMAKE_VERSION VERSION_LESS "3.23.0")
104104
message(STATUS "CMake < 3.23.0; determining CUDA architectures supported...")
105105

106-
# 11.x and 12.x both support these at a minimum.
107-
set(CMAKE_CUDA_ARCHITECTURES_ALL 50 52 53 60 61 62 70 72 75 80)
106+
# 11.4+ supports these at a minimum.
107+
set(CMAKE_CUDA_ARCHITECTURES_ALL 50 52 53 60 61 62 70 72 75 80 86 87)
108108
set(CMAKE_CUDA_ARCHITECTURES_ALL_MAJOR 50 60 70 80)
109109

110-
# CUDA 11.1 adds Ampere support for GA102-GA107.
111-
if (CMAKE_CUDA_COMPILER_VERSION VERSION_GREATER_EQUAL "11.1")
112-
list(APPEND CMAKE_CUDA_ARCHITECTURES_ALL 86)
113-
endif()
114-
115-
# CUDA 11.4 adds Ampere support for GA10B.
116-
if (CMAKE_CUDA_COMPILER_VERSION VERSION_GREATER_EQUAL "11.4")
117-
list(APPEND CMAKE_CUDA_ARCHITECTURES_ALL 87)
118-
endif()
119-
120110
# CUDA 11.8 adds support for Ada and Hopper.
121111
if (CMAKE_CUDA_COMPILER_VERSION VERSION_GREATER_EQUAL "11.8")
122112
list(APPEND CMAKE_CUDA_ARCHITECTURES_ALL 89 90)
123113
list(APPEND CMAKE_CUDA_ARCHITECTURES_ALL_MAJOR 90)
124114
endif()
125-
# CUDA 12.7 adds support for Blackwell B100.
126-
if (CMAKE_CUDA_COMPILER_VERSION VERSION_GREATER_EQUAL "12.7")
127-
list(APPEND CMAKE_CUDA_ARCHITECTURES_ALL 100)
128-
list(APPEND CMAKE_CUDA_ARCHITECTURES_ALL_MAJOR 100)
129-
endif()
130115

131-
# CUDA 12.8 adds support for RTX 50 Blackwell.
116+
# CUDA 12.8 adds support for Blackwell.
132117
if (CMAKE_CUDA_COMPILER_VERSION VERSION_GREATER_EQUAL "12.8")
133-
list(APPEND CMAKE_CUDA_ARCHITECTURES_ALL 101 120)
134-
list(APPEND CMAKE_CUDA_ARCHITECTURES_ALL_MAJOR 101 120)
118+
list(APPEND CMAKE_CUDA_ARCHITECTURES_ALL 100 101 120)
119+
list(APPEND CMAKE_CUDA_ARCHITECTURES_ALL_MAJOR 100 120)
135120
endif()
136121
endif()
137122

0 commit comments

Comments
 (0)