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

Signed-off-by: Tim Moon <[email protected]>
NVIDIA · Oct 14, 2024 · 86f07be · 86f07be
1 parent 55dcbb4
commit 86f07be
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/transformer_engine/common/CMakeLists.txt b/transformer_engine/common/CMakeLists.txt
@@ -88,7 +88,6 @@ target_include_directories(transformer_engine PUBLIC
 # Configure dependencies
 target_link_libraries(transformer_engine PUBLIC
                       CUDA::cublas
-                      CUDA::cuda_driver
                       CUDA::cudart)
 target_include_directories(transformer_engine PRIVATE
                            ${CMAKE_CUDA_TOOLKIT_INCLUDE_DIRECTORIES})