From 2512c1d7df69b4eee4242f007ad1309c45e301e7 Mon Sep 17 00:00:00 2001 From: Matthew Kotila Date: Thu, 24 Oct 2024 08:15:15 -0700 Subject: [PATCH] Remove unnecessary linking of CUDA Runtime library --- src/c++/library/CMakeLists.txt | 12 ++++-------- 1 file changed, 4 insertions(+), 8 deletions(-) diff --git a/src/c++/library/CMakeLists.txt b/src/c++/library/CMakeLists.txt index b5656bfaf..7fa1ca9f7 100644 --- a/src/c++/library/CMakeLists.txt +++ b/src/c++/library/CMakeLists.txt @@ -291,10 +291,8 @@ if(TRITON_ENABLE_CC_GRPC) endif() # TRITON_ENABLE_GPU if(TRITON_ENABLE_GPU) - target_link_libraries( - ${_client_target} - PUBLIC CUDA::cudart - ) + target_include_directories(${_client_target} + PUBLIC ${CUDAToolkit_INCLUDE_DIRS}) endif() # TRITON_ENABLE_GPU endforeach() @@ -477,10 +475,8 @@ if(TRITON_ENABLE_CC_HTTP) endif() # TRITON_ENABLE_GPU if(TRITON_ENABLE_GPU) - target_link_libraries( - ${_client_target} - PUBLIC CUDA::cudart - ) + target_include_directories(${_client_target} + PUBLIC ${CUDAToolkit_INCLUDE_DIRS}) endif() # TRITON_ENABLE_GPU if(${TRITON_ENABLE_ZLIB})