From 673b87a311be5c51d73ba90f30655059c6e9697a Mon Sep 17 00:00:00 2001 From: Concedo <39025047+LostRuins@users.noreply.github.com> Date: Sun, 6 Oct 2024 16:37:08 +0800 Subject: [PATCH] undo header removal --- src/llama.cpp | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/llama.cpp b/src/llama.cpp index b715fa541..6c69381a8 100644 --- a/src/llama.cpp +++ b/src/llama.cpp @@ -15,7 +15,9 @@ # include "ggml-rpc.h" #endif -#if defined(GGML_USE_CLBLAST) +#ifdef GGML_USE_CUDA +# include "ggml-cuda.h" +#elif defined(GGML_USE_CLBLAST) # include "ggml-opencl.h" #elif defined(GGML_USE_VULKAN) # include "ggml-vulkan.h"