From 33721615b5a6513351b07dcefd1094c33337bda6 Mon Sep 17 00:00:00 2001
From: Concedo <39025047+LostRuins@users.noreply.github.com>
Date: Sun, 3 Nov 2024 11:01:51 +0800
Subject: [PATCH] fixed build issues

---
 src/llama.cpp | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/src/llama.cpp b/src/llama.cpp
index 882e90be6..355770826 100644
--- a/src/llama.cpp
+++ b/src/llama.cpp
@@ -12,10 +12,13 @@
 #include "ggml-backend.h"
 #include "ggml-cpp.h"
 
-#if defined(GGML_USE_CLBLAST)
+#ifdef GGML_USE_CUDA
+#  include "ggml-cuda.h"
+#elif defined(GGML_USE_CLBLAST)
 #  include "ggml-opencl.h"
 #endif
 
+
 // TODO: replace with ggml API call
 #define QK_K 256