Unverified Commit da317197 authored by Yuhong Guo's avatar Yuhong Guo Committed by GitHub
Browse files

[Build] Fix cuda link target of cumem_allocator in CPU env (#12863)


Signed-off-by: default avatarYuhongGuo <yuhong.gyh@antgroup.com>
Co-authored-by: default avatarTyler Michael Smith <tyler@neuralmagic.com>
parent 7539bbc6
...@@ -192,7 +192,7 @@ set_gencode_flags_for_srcs( ...@@ -192,7 +192,7 @@ set_gencode_flags_for_srcs(
if(VLLM_GPU_LANG STREQUAL "CUDA") if(VLLM_GPU_LANG STREQUAL "CUDA")
message(STATUS "Enabling cumem allocator extension.") message(STATUS "Enabling cumem allocator extension.")
# link against cuda driver library # link against cuda driver library
list(APPEND CUMEM_LIBS cuda) list(APPEND CUMEM_LIBS CUDA::cuda_driver)
define_gpu_extension_target( define_gpu_extension_target(
cumem_allocator cumem_allocator
DESTINATION vllm DESTINATION vllm
......
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