From ce4a979c9036ff96475f0d74bb499834f8d20b34 Mon Sep 17 00:00:00 2001
From: clibdev <clibdev@gmail.com>
Date: Sat, 17 Feb 2024 11:27:31 +0200
Subject: [PATCH] cmake cuda: fix

---
 CMakeLists.txt | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/CMakeLists.txt b/CMakeLists.txt
index 2a922fdb3..4a47a8a08 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -767,9 +767,10 @@ endif()
 set(CUDA_CXX_FLAGS "")
 
 if (LLAMA_CUBLAS)
-    set(CUDA_FLAGS ${CXX_FLAGS} -use_fast_math)
+    set(CUDA_FLAGS -use_fast_math)
+    list(APPEND CUDA_CXX_FLAGS ${CXX_FLAGS})
     if (NOT MSVC)
-        list(APPEND CUDA_FLAGS -Wno-pedantic)
+        list(APPEND CUDA_CXX_FLAGS -Wno-pedantic)
     endif()
 
     if (LLAMA_ALL_WARNINGS AND NOT MSVC)