Fix cmake config error

Signed-off-by: onepick <jiajuku12@163.com>

Fix cmake config error
Signed-off-by: onepick <jiajuku12@163.com>
c5edd3fd · mykg · 6a7624fe · c5edd3fd
Commit c5edd3fd authored Apr 18, 2025 by mykg
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

csrc/ktransformers_ext/CMakeLists.txt csrc/ktransformers_ext/CMakeLists.txt +2 -0

No files found.
--- a/csrc/ktransformers_ext/CMakeLists.txt
+++ b/csrc/ktransformers_ext/CMakeLists.txt
@@ -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