Skip to content

Commit 19b55ad

Browse files
authored
feat: use gpu backend for clip if available (abetlen#1175)
1 parent 63b0c37 commit 19b55ad

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

CMakeLists.txt

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -46,6 +46,14 @@ if (LLAMA_BUILD)
4646
)
4747

4848
if (LLAVA_BUILD)
49+
if (LLAMA_CUBLAS)
50+
add_compile_definitions(GGML_USE_CUBLAS)
51+
endif()
52+
53+
if (LLAMA_METAL)
54+
add_compile_definitions(GGML_USE_METAL)
55+
endif()
56+
4957
# Building llava
5058
add_subdirectory(vendor/llama.cpp/examples/llava)
5159
set_target_properties(llava_shared PROPERTIES OUTPUT_NAME "llava")

0 commit comments

Comments
 (0)