cmake : set default CUDA architectures (#1667)

diff --git a/CMakeLists.txt b/CMakeLists.txt
index 03e6bd01c2a052dd8631c2ecd6383eb1ffebeb72..b6bbc044fc8aba7a7a18b18386832f6dad6f0f4d 100644 (file)
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -521,7 +521,13 @@ endif()
 
 if (GGML_SOURCES_CUDA)
     message(STATUS "GGML CUDA sources found, configuring CUDA architecture")
-    set_property(TARGET whisper PROPERTY CUDA_ARCHITECTURES OFF)
+    # Only configure gmml CUDA architectures is not globally set
+    if (NOT DEFINED GGML_CUDA_ARCHITECTURES)
+        # Not overriden by user, so set defaults
+        set(GGML_CUDA_ARCHITECTURES 52 61 70)
+    endif()
+    message(STATUS "GGML Configuring CUDA architectures ${GGML_CUDA_ARCHITECTURES}")
+    set_property(TARGET whisper PROPERTY CUDA_ARCHITECTURES ${GGML_CUDA_ARCHITECTURES})
     set_property(TARGET whisper PROPERTY CUDA_SELECT_NVCC_ARCH_FLAGS "Auto")
 endif()