We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 5d6cbd0 commit b74bb57Copy full SHA for b74bb57
CMakeLists.txt
@@ -446,6 +446,9 @@ if(VLLM_GPU_LANG STREQUAL "CUDA")
446
endif()
447
448
message(STATUS "Enabling C extension.")
449
+if(VLLM_GPU_LANG STREQUAL "CUDA")
450
+ list(APPEND VLLM_C_LIBS cuda)
451
+endif()
452
define_gpu_extension_target(
453
_C
454
DESTINATION vllm
@@ -454,6 +457,7 @@ define_gpu_extension_target(
457
COMPILE_FLAGS ${VLLM_GPU_FLAGS}
455
458
ARCHITECTURES ${VLLM_GPU_ARCHES}
456
459
INCLUDE_DIRECTORIES ${CUTLASS_INCLUDE_DIR};${CUTLASS_TOOLS_UTIL_INCLUDE_DIR}
460
+ LIBRARIES ${VLLM_C_LIBS}
461
USE_SABI 3
462
WITH_SOABI)
463
0 commit comments