Skip to content

Commit 12a0d43

Browse files
authored
Update CMakeLists.txt
1 parent 3e9dedb commit 12a0d43

1 file changed

Lines changed: 13 additions & 0 deletions

File tree

CMakeLists.txt

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -168,6 +168,19 @@ if(BUILD_CUDA)
168168
list(APPEND POSSIBLE_CAPABILITIES ${capability_id})
169169
endif()
170170
endforeach()
171+
172+
# CUDA 13+: drop <=80 and add new SMs
173+
if (CMAKE_CUDA_COMPILER_VERSION VERSION_GREATER_EQUAL "13.0")
174+
list(APPEND CMAKE_CUDA_ARCHITECTURES_ALL 103 110 121)
175+
list(REMOVE_ITEM CMAKE_CUDA_ARCHITECTURES_ALL 50 52 53 60 61 62 70 72 75)
176+
list(REMOVE_ITEM CMAKE_CUDA_ARCHITECTURES_ALL_MAJOR 50 60 70)
177+
list(APPEND CMAKE_CUDA_ARCHITECTURES_ALL_MAJOR 80 90 100 120)
178+
endif()
179+
180+
list(REMOVE_DUPLICATES CMAKE_CUDA_ARCHITECTURES_ALL)
181+
list(REMOVE_DUPLICATES CMAKE_CUDA_ARCHITECTURES_ALL_MAJOR)
182+
list(SORT CMAKE_CUDA_ARCHITECTURES_ALL COMPARE NATURAL)
183+
list(SORT CMAKE_CUDA_ARCHITECTURES_ALL_MAJOR COMPARE NATURAL)
171184

172185
# This can be changed via -D argument to CMake
173186
# By default all possible capabilities are compiled

0 commit comments

Comments
 (0)