Skip to content

GitLab

"git@developer.sourcefind.cn:OpenDAS/ollama.git" did not exist on "74bd09652d69c77a4bed34b3afda74c87295115b"
Unverified Commit 34c19940 authored by Atream's avatar Atream Committed by GitHub
Browse files

Merge pull request #1170 from onepick/fix-cmake-error 

Fix cmake config error
parents 0892d37d c5edd3fd
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
csrc/ktransformers_ext/CMakeLists.txt
View file @ 34c19940
...@@ -285,6 +285,7 @@ elseif (UNIX) ...@@ -285,6 +285,7 @@ elseif (UNIX)
message(STATUS "enabling CUDA") message(STATUS "enabling CUDA")
enable_language(CUDA) enable_language(CUDA)
add_compile_definitions(KTRANSFORMERS_USE_CUDA=1) add_compile_definitions(KTRANSFORMERS_USE_CUDA=1)
endif()
endif() endif()
aux_source_directory(${CMAKE_CURRENT_SOURCE_DIR} SOURCE_DIR1) aux_source_directory(${CMAKE_CURRENT_SOURCE_DIR} SOURCE_DIR1)
...@@ -327,6 +328,7 @@ elseif(UNIX) ...@@ -327,6 +328,7 @@ elseif(UNIX)
target_link_libraries(${PROJECT_NAME} PRIVATE MUSA::musart) target_link_libraries(${PROJECT_NAME} PRIVATE MUSA::musart)
else() else()
target_link_libraries(${PROJECT_NAME} PRIVATE "${CUDAToolkit_LIBRARY_DIR}/libcudart.so") target_link_libraries(${PROJECT_NAME} PRIVATE "${CUDAToolkit_LIBRARY_DIR}/libcudart.so")
endif()
endif() endif()
# Define the USE_NUMA option # Define the USE_NUMA option
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment