Skip to content

Commit 72bac73

Browse files
authored
[Build/CI] Fix libcuda.so linkage (#12424)
Signed-off-by: Tyler Michael Smith <[email protected]>
1 parent 68f1114 commit 72bac73

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

CMakeLists.txt

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -446,6 +446,9 @@ if(VLLM_GPU_LANG STREQUAL "CUDA")
446446
endif()
447447

448448
message(STATUS "Enabling C extension.")
449+
if(VLLM_GPU_LANG STREQUAL "CUDA")
450+
list(APPEND VLLM_C_LIBS cuda)
451+
endif()
449452
define_gpu_extension_target(
450453
_C
451454
DESTINATION vllm
@@ -454,6 +457,7 @@ define_gpu_extension_target(
454457
COMPILE_FLAGS ${VLLM_GPU_FLAGS}
455458
ARCHITECTURES ${VLLM_GPU_ARCHES}
456459
INCLUDE_DIRECTORIES ${CUTLASS_INCLUDE_DIR};${CUTLASS_TOOLS_UTIL_INCLUDE_DIR}
460+
LIBRARIES ${VLLM_C_LIBS}
457461
USE_SABI 3
458462
WITH_SOABI)
459463

0 commit comments

Comments
 (0)