vrr/kvcache-ai-ktransformers
2
0
Fork 0
mirror of https://github.com/kvcache-ai/ktransformers.git synced 2026-05-05 23:50:14 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

Merge pull request #1170 from onepick/fix-cmake-error
Some checks failed
Deploy / deploy (ubuntu-latest) (push) Failing after 3s
Details
Deploy / deploy (windows-latest) (push) Has been cancelled
Details
Book-CI / test (push) Has been cancelled
Details
Deploy / deploy (macos-latest) (push) Has been cancelled
Details

Browse source 
Fix cmake config error
This commit is contained in:
Atream 2025-04-18 07:51:03 -06:00 • committed by GitHub
commit 34c199403b
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 2 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

2
csrc/ktransformers_ext/CMakeLists.txt
View file

@ -285,6 +285,7 @@ elseif (UNIX)
message(STATUS "enabling CUDA")
enable_language(CUDA)
add_compile_definitions(KTRANSFORMERS_USE_CUDA=1)
endif()
endif()
aux_source_directory(${CMAKE_CURRENT_SOURCE_DIR} SOURCE_DIR1)
@ -327,6 +328,7 @@ elseif(UNIX)
target_link_libraries(${PROJECT_NAME} PRIVATE MUSA::musart)
else()
target_link_libraries(${PROJECT_NAME} PRIVATE "${CUDAToolkit_LIBRARY_DIR}/libcudart.so")
endif()
endif()
# Define the USE_NUMA option