Skip to content

Commit 91e8414

Browse files
authored
fix(cmake): logic detect sm70, sm7.5 (#4175)
1 parent 4ca67f5 commit 91e8414

1 file changed

Lines changed: 3 additions & 2 deletions

File tree

CMakeLists.txt

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -220,8 +220,9 @@ endif()
220220

221221
if(ARCH STREQUAL "x86_64")
222222
if (NOT CMAKE_CUDA_ARCHITECTURES)
223-
if (${CMAKE_CUDA_COMPILER_VERSION} VERSION_LESS_EQUAL "12")
224-
set(CMAKE_CUDA_ARCHITECTURES 70-real 75-real) # V100, 2080
223+
set(CMAKE_CUDA_ARCHITECTURES "")
224+
if (${CMAKE_CUDA_COMPILER_VERSION} VERSION_LESS "13.0")
225+
list(APPEND CMAKE_CUDA_ARCHITECTURES 70-real 75-real) # V100, 2080
225226
endif()
226227
if (${CMAKE_CUDA_COMPILER_VERSION} VERSION_GREATER_EQUAL "11")
227228
list(APPEND CMAKE_CUDA_ARCHITECTURES 80-real) # A100

0 commit comments

Comments
 (0)