Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add HIP LIB PATH

Browse Source
This commit is contained in:
Hui Liu 2023-10-23 16:52:33 -07:00
parent 643b334919
commit f9f39b092b
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
cmake/Utils.cmake
View File

@ -202,13 +202,15 @@ endmacro()
macro(xgboost_link_rccl target) macro(xgboost_link_rccl target)
if(BUILD_STATIC_LIB) 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_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() 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_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() endif()
endmacro() endmacro()