limit to GGML_ALLOW_CUDA_GRAPHS defined in llama.cpp cmake

This commit is contained in:
Alan Gray 2024-04-25 00:51:48 -07:00
parent 408759687f
commit 0640427f7b
2 changed files with 2 additions and 1 deletions

View file

@ -413,6 +413,7 @@ if (LLAMA_CUDA)
list(APPEND GGML_SOURCES_CUDA "ggml-cuda.cu")
add_compile_definitions(GGML_USE_CUDA)
add_compile_definitions(GGML_ALLOW_CUDA_GRAPHS)
if (LLAMA_CUDA_FORCE_DMMV)
add_compile_definitions(GGML_CUDA_FORCE_DMMV)
endif()