summaryrefslogtreecommitdiff
path: root/examples/llava/clip.cpp
diff options
context:
space:
mode:
authorslaren <slarengh@gmail.com>2024-03-26 01:16:01 +0100
committerGitHub <noreply@github.com>2024-03-26 01:16:01 +0100
commit280345968dabc00d212d43e31145f5c9961a7604 (patch)
tree4d0ada8b59a4c15cb6d4fe1a6b4740a30dcdb0f2 /examples/llava/clip.cpp
parentb06c16ef9f81d84da520232c125d4d8a1d273736 (diff)
cuda : rename build flag to LLAMA_CUDA (#6299)
Diffstat (limited to 'examples/llava/clip.cpp')
-rw-r--r--examples/llava/clip.cpp4
1 files changed, 2 insertions, 2 deletions
diff --git a/examples/llava/clip.cpp b/examples/llava/clip.cpp
index 48caafa8..40c97626 100644
--- a/examples/llava/clip.cpp
+++ b/examples/llava/clip.cpp
@@ -7,7 +7,7 @@
#include "ggml-alloc.h"
#include "ggml-backend.h"
-#ifdef GGML_USE_CUBLAS
+#ifdef GGML_USE_CUDA
#include "ggml-cuda.h"
#endif
@@ -968,7 +968,7 @@ struct clip_ctx * clip_model_load(const char * fname, const int verbosity = 1) {
}
}
-#ifdef GGML_USE_CUBLAS
+#ifdef GGML_USE_CUDA
new_clip->backend = ggml_backend_cuda_init(0);
printf("%s: CLIP using CUDA backend\n", __func__);
#endif