Skip to content

GitLab

Commit c5edd3fd authored by mykg's avatar mykg
Browse files

Fix cmake config error 


Signed-off-by: mykg's avataronepick <jiajuku12@163.com>
parent 6a7624fe
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
csrc/ktransformers_ext/CMakeLists.txt
View file @ c5edd3fd
...@@ -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