Revert "Remove unnecessary linking of CUDA Runtime library (#798)"

This reverts commit 51a5e91.
triton-inference-server · Oct 25, 2024 · 4321c37 · 4321c37
1 parent 51a5e91
commit 4321c37
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/src/c++/library/CMakeLists.txt b/src/c++/library/CMakeLists.txt
@@ -291,8 +291,10 @@ if(TRITON_ENABLE_CC_GRPC)
     endif() # TRITON_ENABLE_GPU
 
     if(TRITON_ENABLE_GPU)
-      target_include_directories(${_client_target}
-                                 PUBLIC ${CUDAToolkit_INCLUDE_DIRS})
+      target_link_libraries(
+        ${_client_target}
+        PUBLIC CUDA::cudart
+      )
     endif() # TRITON_ENABLE_GPU
   endforeach()
 
@@ -475,8 +477,10 @@ if(TRITON_ENABLE_CC_HTTP)
     endif() # TRITON_ENABLE_GPU
 
     if(TRITON_ENABLE_GPU)
-      target_include_directories(${_client_target}
-                                 PUBLIC ${CUDAToolkit_INCLUDE_DIRS})
+      target_link_libraries(
+        ${_client_target}
+        PUBLIC CUDA::cudart
+      )
     endif() # TRITON_ENABLE_GPU
 
     if(${TRITON_ENABLE_ZLIB})