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

add hip

Browse Source
This commit is contained in:
amdsc21 2023-03-08 03:32:19 +01:00
parent 840f15209c
commit 52b05d934e
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
cmake/xgboost-config.cmake.in
View File

@ -3,6 +3,8 @@
set(USE_OPENMP @USE_OPENMP@) set(USE_OPENMP @USE_OPENMP@)
set(USE_CUDA @USE_CUDA@) set(USE_CUDA @USE_CUDA@)
set(USE_NCCL @USE_NCCL@) set(USE_NCCL @USE_NCCL@)
set(USE_HIP @USE_HIP@)
set(USE_RCCL @USE_RCCL@)
set(XGBOOST_BUILD_STATIC_LIB @BUILD_STATIC_LIB@) set(XGBOOST_BUILD_STATIC_LIB @BUILD_STATIC_LIB@)
include(CMakeFindDependencyMacro) include(CMakeFindDependencyMacro)
@ -15,6 +17,9 @@ if (XGBOOST_BUILD_STATIC_LIB)
if(USE_CUDA) if(USE_CUDA)
find_dependency(CUDA) find_dependency(CUDA)
endif() endif()
if(USE_HIP)
find_dependency(HIP)
endif()
# nccl should be linked statically if xgboost is built as static library. # nccl should be linked statically if xgboost is built as static library.
endif (XGBOOST_BUILD_STATIC_LIB) endif (XGBOOST_BUILD_STATIC_LIB)