Skip to content

Commit 590fadb

Browse files
gjasnyggerganov
authored andcommitted
CUDA : conditionally add cuda architectures (ggml/1341)
1 parent 803dac2 commit 590fadb

File tree

1 file changed

+7
-3
lines changed

1 file changed

+7
-3
lines changed

ggml/src/ggml-cuda/CMakeLists.txt

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -25,10 +25,14 @@ if (CUDAToolkit_FOUND)
2525
if (GGML_NATIVE AND CUDAToolkit_VERSION VERSION_GREATER_EQUAL "11.6" AND CMAKE_VERSION VERSION_GREATER_EQUAL "3.24")
2626
set(CMAKE_CUDA_ARCHITECTURES "native")
2727
else()
28+
if (CUDAToolkit_VERSION VERSION_LESS "13")
29+
list(APPEND CMAKE_CUDA_ARCHITECTURES 50-virtual 61-virtual 70-virtual)
30+
endif ()
31+
32+
list(APPEND CMAKE_CUDA_ARCHITECTURES 75-virtual 80-virtual 86-real)
33+
2834
if (CUDAToolkit_VERSION VERSION_GREATER_EQUAL "11.8")
29-
set(CMAKE_CUDA_ARCHITECTURES "50-virtual;61-virtual;70-virtual;75-virtual;80-virtual;86-real;89-real")
30-
else()
31-
set(CMAKE_CUDA_ARCHITECTURES "50-virtual;61-virtual;70-virtual;75-virtual;80-virtual;86-real")
35+
list(APPEND CMAKE_CUDA_ARCHITECTURES 89-real)
3236
endif()
3337
endif()
3438
endif()

0 commit comments

Comments
 (0)