Fix cuda link target of cumem_allocator

Signed-off-by: YuhongGuo <yuhong.gyh@antgroup.com>
vllm-project · Feb 7, 2025 · 3b09f0c · 3b09f0c
1 parent 56534cd
commit 3b09f0c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -192,7 +192,7 @@ set_gencode_flags_for_srcs(
 if(VLLM_GPU_LANG STREQUAL "CUDA")
   message(STATUS "Enabling cumem allocator extension.")
   # link against cuda driver library
-  list(APPEND CUMEM_LIBS cuda)
+  list(APPEND CUMEM_LIBS CUDA::cuda_driver)
   define_gpu_extension_target(
     cumem_allocator
     DESTINATION vllm