|
@@ -15,9 +15,9 @@ if (CUDAToolkit_FOUND)
|
|
|
if (GGML_NATIVE AND CUDAToolkit_VERSION VERSION_GREATER_EQUAL "11.6" AND CMAKE_VERSION VERSION_GREATER_EQUAL "3.24")
|
|
if (GGML_NATIVE AND CUDAToolkit_VERSION VERSION_GREATER_EQUAL "11.6" AND CMAKE_VERSION VERSION_GREATER_EQUAL "3.24")
|
|
|
set(CMAKE_CUDA_ARCHITECTURES "native")
|
|
set(CMAKE_CUDA_ARCHITECTURES "native")
|
|
|
elseif(GGML_CUDA_F16 OR GGML_CUDA_DMMV_F16)
|
|
elseif(GGML_CUDA_F16 OR GGML_CUDA_DMMV_F16)
|
|
|
- set(CMAKE_CUDA_ARCHITECTURES "60;61;70;75")
|
|
|
|
|
|
|
+ set(CMAKE_CUDA_ARCHITECTURES "60;61;70;75;80")
|
|
|
else()
|
|
else()
|
|
|
- set(CMAKE_CUDA_ARCHITECTURES "52;61;70;75")
|
|
|
|
|
|
|
+ set(CMAKE_CUDA_ARCHITECTURES "52;61;70;75;80")
|
|
|
endif()
|
|
endif()
|
|
|
endif()
|
|
endif()
|
|
|
message(STATUS "Using CUDA architectures: ${CMAKE_CUDA_ARCHITECTURES}")
|
|
message(STATUS "Using CUDA architectures: ${CMAKE_CUDA_ARCHITECTURES}")
|