Unverified Commit 86f07be4 authored by Tim Moon's avatar Tim Moon Committed by GitHub
Browse files

Do not link against CUDA driver when building (#1240)


Signed-off-by: default avatarTim Moon <tmoon@nvidia.com>
parent 55dcbb4b
...@@ -88,7 +88,6 @@ target_include_directories(transformer_engine PUBLIC ...@@ -88,7 +88,6 @@ target_include_directories(transformer_engine PUBLIC
# Configure dependencies # Configure dependencies
target_link_libraries(transformer_engine PUBLIC target_link_libraries(transformer_engine PUBLIC
CUDA::cublas CUDA::cublas
CUDA::cuda_driver
CUDA::cudart) CUDA::cudart)
target_include_directories(transformer_engine PRIVATE target_include_directories(transformer_engine PRIVATE
${CMAKE_CUDA_TOOLKIT_INCLUDE_DIRECTORIES}) ${CMAKE_CUDA_TOOLKIT_INCLUDE_DIRECTORIES})
......
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