From: Diego Devesa <redacted>
Date: Fri, 14 Feb 2025 14:33:52 +0000 (+0100)
Subject: cuda : add ampere to the list of default architectures (llama/11870)
X-Git-Tag: upstream/1.7.4+203~68
X-Git-Url: https://git.djapps.eu/?a=commitdiff_plain;h=47cc043e69e0493a74336b2383603479dcbf2ea5;p=pkg%2Fggml%2Fsources%2Fwhisper.cpp

cuda : add ampere to the list of default architectures (llama/11870)
---

diff --git a/ggml/src/ggml-cuda/CMakeLists.txt b/ggml/src/ggml-cuda/CMakeLists.txt
index 119fd39b..682640b5 100644
--- a/ggml/src/ggml-cuda/CMakeLists.txt
+++ b/ggml/src/ggml-cuda/CMakeLists.txt
@@ -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")
             set(CMAKE_CUDA_ARCHITECTURES "native")
         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()
-            set(CMAKE_CUDA_ARCHITECTURES "52;61;70;75")
+            set(CMAKE_CUDA_ARCHITECTURES "52;61;70;75;80")
         endif()
     endif()
     message(STATUS "Using CUDA architectures: ${CMAKE_CUDA_ARCHITECTURES}")