From: Borislav Stanimirov <redacted>
Date: Mon, 30 Oct 2023 04:28:11 +0000 (+0200)
Subject: cmake : cuda architectures: allow user override, only set local if not globally set... 
X-Git-Tag: upstream/0.0.1642~1209
X-Git-Url: https://git.djapps.eu/?a=commitdiff_plain;h=306f162115c3b66aaa687fd7ae88e913a63f5d24;p=pkg%2Fggml%2Fsources%2Fggml

cmake : cuda architectures: allow user override, only set local if not globally set (#595)
---

diff --git a/src/CMakeLists.txt b/src/CMakeLists.txt
index b225597e..50c0b92c 100644
--- a/src/CMakeLists.txt
+++ b/src/CMakeLists.txt
@@ -305,8 +305,16 @@ if (MINGW)
 endif()
 
 if (GGML_CUDA_SOURCES)
-    message(STATUS "GGML CUDA sources found, configuring CUDA architecture")
-    set_property(TARGET ggml  PROPERTY CUDA_ARCHITECTURES "52;61;70")
+    message(STATUS "GGML CUDA sources found")
+    if (NOT DEFINED CMAKE_CUDA_ARCHITECTURES)
+        # 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 ggml  PROPERTY CUDA_ARCHITECTURES ${GGML_CUDA_ARCHITECTURES})
+    endif()
     set_property(TARGET ggml  PROPERTY CUDA_SELECT_NVCC_ARCH_FLAGS "Auto")
     if (NOT MSVC)
         target_link_libraries(ggml PUBLIC stdc++)