Skip to content

Commit 94b87f8

Browse files
authored
cuda : add ampere to the list of default architectures (ggml-org#11870)
1 parent dbc2ec5 commit 94b87f8

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

ggml/src/ggml-cuda/CMakeLists.txt

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -15,9 +15,9 @@ if (CUDAToolkit_FOUND)
1515
if (GGML_NATIVE AND CUDAToolkit_VERSION VERSION_GREATER_EQUAL "11.6" AND CMAKE_VERSION VERSION_GREATER_EQUAL "3.24")
1616
set(CMAKE_CUDA_ARCHITECTURES "native")
1717
elseif(GGML_CUDA_F16 OR GGML_CUDA_DMMV_F16)
18-
set(CMAKE_CUDA_ARCHITECTURES "60;61;70;75")
18+
set(CMAKE_CUDA_ARCHITECTURES "60;61;70;75;80")
1919
else()
20-
set(CMAKE_CUDA_ARCHITECTURES "52;61;70;75")
20+
set(CMAKE_CUDA_ARCHITECTURES "52;61;70;75;80")
2121
endif()
2222
endif()
2323
message(STATUS "Using CUDA architectures: ${CMAKE_CUDA_ARCHITECTURES}")

0 commit comments

Comments
 (0)