From f9f39b092ba509b0dc56abb43066af90a00f9662 Mon Sep 17 00:00:00 2001
From: Hui Liu <96135754+amdsc21@users.noreply.github.com>
Date: Mon, 23 Oct 2023 16:52:33 -0700
Subject: [PATCH] add HIP LIB PATH

---
 cmake/Utils.cmake | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/cmake/Utils.cmake b/cmake/Utils.cmake
index ca5c522e1..da4c9a5d8 100644
--- a/cmake/Utils.cmake
+++ b/cmake/Utils.cmake
@@ -202,13 +202,15 @@ endmacro()
 
 macro(xgboost_link_rccl target)
   if(BUILD_STATIC_LIB)
-    target_include_directories(${target} PUBLIC ${rccl_INCLUDE_DIR})
+    target_include_directories(${target} PUBLIC ${RCCL_INCLUDE_DIR})
     target_compile_definitions(${target} PUBLIC -DXGBOOST_USE_RCCL=1)
-    target_link_libraries(${target} PUBLIC ${rccl_LIBRARY})
+    target_link_directories(${target} PUBLIC ${HIP_LIB_INSTALL_DIR})
+    target_link_libraries(${target} PUBLIC ${RCCL_LIBRARY})
   else()
-    target_include_directories(${target} PRIVATE ${rccl_INCLUDE_DIR})
+    target_include_directories(${target} PRIVATE ${RCCL_INCLUDE_DIR})
     target_compile_definitions(${target} PRIVATE -DXGBOOST_USE_RCCL=1)
-    target_link_libraries(${target} PRIVATE ${rccl_LIBRARY})
+    target_link_directories(${target} PUBLIC ${HIP_LIB_INSTALL_DIR})
+    target_link_libraries(${target} PRIVATE ${RCCL_LIBRARY})
   endif()
 endmacro()