Skip to content

Commit da31719

Browse files
[Build] Fix cuda link target of cumem_allocator in CPU env (#12863)
Signed-off-by: YuhongGuo <[email protected]> Co-authored-by: Tyler Michael Smith <[email protected]>
1 parent 7539bbc commit da31719

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

CMakeLists.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -192,7 +192,7 @@ set_gencode_flags_for_srcs(
192192
if(VLLM_GPU_LANG STREQUAL "CUDA")
193193
message(STATUS "Enabling cumem allocator extension.")
194194
# link against cuda driver library
195-
list(APPEND CUMEM_LIBS cuda)
195+
list(APPEND CUMEM_LIBS CUDA::cuda_driver)
196196
define_gpu_extension_target(
197197
cumem_allocator
198198
DESTINATION vllm

0 commit comments

Comments
 (0)