Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
OpenDAS
vllm_cscc
Commits
de6f90a1
Unverified
Commit
de6f90a1
authored
Sep 19, 2024
by
bnellnm
Committed by
GitHub
Sep 20, 2024
Browse files
[Misc] guard against change in cuda library name (#8609)
parent
6cb748e1
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
4 additions
and
3 deletions
+4
-3
cmake/utils.cmake
cmake/utils.cmake
+4
-3
No files found.
cmake/utils.cmake
View file @
de6f90a1
...
...
@@ -350,13 +350,14 @@ function (define_gpu_extension_target GPU_MOD_NAME)
target_include_directories
(
${
GPU_MOD_NAME
}
PRIVATE csrc
${
GPU_INCLUDE_DIRECTORIES
}
)
# TODO: is torch_python_LIBRARY needed?
target_link_libraries
(
${
GPU_MOD_NAME
}
PRIVATE torch
${
torch_python_LIBRARY
}
${
GPU_LIBRARIES
}
)
target_link_libraries
(
${
GPU_MOD_NAME
}
PRIVATE torch
${
GPU_LIBRARIES
}
)
# Don't use `TORCH_LIBRARIES` for CUDA since it pulls in a bunch of
# dependencies that are not necessary and may not be installed.
if
(
GPU_LANGUAGE STREQUAL
"CUDA"
)
if
(
"
${
CUDA_CUDA_LIB
}
"
STREQUAL
""
)
set
(
CUDA_CUDA_LIB
"
${
CUDA_CUDA_LIBRARY
}
"
)
endif
()
target_link_libraries
(
${
GPU_MOD_NAME
}
PRIVATE
${
CUDA_CUDA_LIB
}
${
CUDA_LIBRARIES
}
)
else
()
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment