Skip to content

GitLab

Unverified Commit 72bac730 authored by Tyler Michael Smith's avatar Tyler Michael Smith Committed by GitHub
Browse files

[Build/CI] Fix libcuda.so linkage (#12424) 


Signed-off-by: default avatarTyler Michael Smith <tyler@neuralmagic.com>
parent 68f11149
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 0 deletions
CMakeLists.txt
View file @ 72bac730
...@@ -446,6 +446,9 @@ if(VLLM_GPU_LANG STREQUAL "CUDA") ...@@ -446,6 +446,9 @@ if(VLLM_GPU_LANG STREQUAL "CUDA")
endif() endif()
message(STATUS "Enabling C extension.") message(STATUS "Enabling C extension.")
if(VLLM_GPU_LANG STREQUAL "CUDA")
list(APPEND VLLM_C_LIBS cuda)
endif()
define_gpu_extension_target( define_gpu_extension_target(
_C _C
DESTINATION vllm DESTINATION vllm
...@@ -454,6 +457,7 @@ define_gpu_extension_target( ...@@ -454,6 +457,7 @@ define_gpu_extension_target(
COMPILE_FLAGS ${VLLM_GPU_FLAGS} COMPILE_FLAGS ${VLLM_GPU_FLAGS}
ARCHITECTURES ${VLLM_GPU_ARCHES} ARCHITECTURES ${VLLM_GPU_ARCHES}
INCLUDE_DIRECTORIES ${CUTLASS_INCLUDE_DIR};${CUTLASS_TOOLS_UTIL_INCLUDE_DIR} INCLUDE_DIRECTORIES ${CUTLASS_INCLUDE_DIR};${CUTLASS_TOOLS_UTIL_INCLUDE_DIR}
LIBRARIES ${VLLM_C_LIBS}
USE_SABI 3 USE_SABI 3
WITH_SOABI) WITH_SOABI)
......
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