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