From: Georgi Gerganov <redacted>
Date: Sun, 16 Jun 2024 10:05:11 +0000 (+0300)
Subject: cmake : fix cuda vars (#0)
X-Git-Tag: upstream/0.0.1642~581
X-Git-Url: https://git.djapps.eu/?a=commitdiff_plain;h=49c77e6805ab8cb2f6cb37b4f99ac540597eba30;p=pkg%2Fggml%2Fsources%2Fggml

cmake : fix cuda vars (#0)
---

diff --git a/src/CMakeLists.txt b/src/CMakeLists.txt
index fdd4d586..25fb1ad1 100644
--- a/src/CMakeLists.txt
+++ b/src/CMakeLists.txt
@@ -194,7 +194,7 @@ if (GGML_CUDA)
             # 60 == f16 CUDA intrinsics
             # 61 == integer CUDA intrinsics
             # 70 == compute capability at which unrolling a loop in mul_mat_q kernels is faster
-            if (LLAMA_CUDA_F16 OR LLAMA_CUDA_DMMV_F16)
+            if (GGML_CUDA_F16 OR GGML_CUDA_DMMV_F16)
                 set(CMAKE_CUDA_ARCHITECTURES "60;61;70") # needed for f16 CUDA intrinsics
             else()
                 set(CMAKE_CUDA_ARCHITECTURES "52;61;70") # lowest CUDA 12 standard + lowest for integer intrinsics