diff options
author | Howard Su <howard0su@gmail.com> | 2023-06-19 23:10:37 +0800 |
---|---|---|
committer | GitHub <noreply@github.com> | 2023-06-19 18:10:37 +0300 |
commit | 1e3abfcef073e73c2b31e8570cb06c5cb2fd1f55 (patch) | |
tree | d2042362c982f38df70c8b24aa5f42de49b636fb | |
parent | 16b9cd193965769089881bb8ec012fccca7b37b6 (diff) |
cmake : fix build shared ggml when CUDA is enabled (#1929)
Co-authored-by: Georgi Gerganov <ggerganov@gmail.com>
-rw-r--r-- | CMakeLists.txt | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/CMakeLists.txt b/CMakeLists.txt index dc06365..a598593 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -469,6 +469,7 @@ add_library(ggml_static STATIC $<TARGET_OBJECTS:ggml>) if (BUILD_SHARED_LIBS) set_target_properties(ggml PROPERTIES POSITION_INDEPENDENT_CODE ON) add_library(ggml_shared SHARED $<TARGET_OBJECTS:ggml>) + target_link_libraries(ggml_shared PUBLIC Threads::Threads ${LLAMA_EXTRA_LIBS}) endif() add_library(llama @@ -500,6 +501,11 @@ if (GGML_SOURCES_CUDA) set_property(TARGET ggml_static PROPERTY CUDA_ARCHITECTURES "native") set_property(TARGET ggml_static PROPERTY CUDA_SELECT_NVCC_ARCH_FLAGS "Auto") + if (BUILD_SHARED_LIBS) + set_property(TARGET ggml_shared PROPERTY CUDA_ARCHITECTURES "native") + set_property(TARGET ggml_shared PROPERTY CUDA_SELECT_NVCC_ARCH_FLAGS "Auto") + endif() + set_property(TARGET llama PROPERTY CUDA_ARCHITECTURES "native") endif() |