File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments