[Build] Fix cuda link target of cumem_allocator in CPU env (vllm-proj…

…ect#12863) Signed-off-by: YuhongGuo <[email protected]> Co-authored-by: Tyler Michael Smith <[email protected]>
jeejeelee · Feb 11, 2025 · da31719 · da31719
1 parent 7539bbc
commit da31719
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