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

fix uuid and Clear/SetValid

Browse Source
This commit is contained in:
Hui Liu
2023-10-23 16:32:26 -07:00
parent 55994b1ac7
commit 6ba66463b6
8 changed files with 44 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
cmake/Utils.cmake
View File

@@ -200,6 +200,18 @@ macro(xgboost_link_nccl target)
endif()
endmacro()
macro(xgboost_link_rccl target)
if(BUILD_STATIC_LIB)
target_include_directories(${target} PUBLIC ${rccl_INCLUDE_DIR})
target_compile_definitions(${target} PUBLIC -DXGBOOST_USE_RCCL=1)
target_link_libraries(${target} PUBLIC ${rccl_LIBRARY})
else()
target_include_directories(${target} PRIVATE ${rccl_INCLUDE_DIR})
target_compile_definitions(${target} PRIVATE -DXGBOOST_USE_RCCL=1)
target_link_libraries(${target} PRIVATE ${rccl_LIBRARY})
endif()
endmacro()
# compile options
macro(xgboost_target_properties target)
set_target_properties(${target} PROPERTIES
@@ -302,6 +314,10 @@ macro(xgboost_target_link_libraries target)
xgboost_link_nccl(${target})
endif()
if(USE_RCCL)
xgboost_link_rccl(${target})
endif()
if(USE_NVTX)
target_link_libraries(${target} PRIVATE CUDA::nvToolsExt)
endif()