We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent dbc2ec5 commit 94b87f8Copy full SHA for 94b87f8
ggml/src/ggml-cuda/CMakeLists.txt
@@ -15,9 +15,9 @@ if (CUDAToolkit_FOUND)
15
if (GGML_NATIVE AND CUDAToolkit_VERSION VERSION_GREATER_EQUAL "11.6" AND CMAKE_VERSION VERSION_GREATER_EQUAL "3.24")
16
set(CMAKE_CUDA_ARCHITECTURES "native")
17
elseif(GGML_CUDA_F16 OR GGML_CUDA_DMMV_F16)
18
- set(CMAKE_CUDA_ARCHITECTURES "60;61;70;75")
+ set(CMAKE_CUDA_ARCHITECTURES "60;61;70;75;80")
19
else()
20
- set(CMAKE_CUDA_ARCHITECTURES "52;61;70;75")
+ set(CMAKE_CUDA_ARCHITECTURES "52;61;70;75;80")
21
endif()
22
23
message(STATUS "Using CUDA architectures: ${CMAKE_CUDA_ARCHITECTURES}")
0 commit comments