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

add hip to config

Browse Source
This commit is contained in:
Hui Liu 2023-10-30 12:01:24 -07:00
parent 8d160a206e
commit d0774a78e4
2 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
R-package/src/xgboost_custom.cc
View File

@ -32,7 +32,7 @@ namespace common {
bool CheckNAN(double v) { bool CheckNAN(double v) {
return ISNAN(v); return ISNAN(v);
} }
#if !defined(XGBOOST_USE_CUDA) #if !defined(XGBOOST_USE_CUDA) && !defined(XGBOOST_USE_HIP)
double LogGamma(double v) { double LogGamma(double v) {
return lgammafn(v); return lgammafn(v);
} }

4
python-package/packager/build_config.py
View File

@ -15,6 +15,10 @@ class BuildConfiguration: # pylint: disable=R0902
use_cuda: bool = False use_cuda: bool = False
# Whether to enable NCCL # Whether to enable NCCL
use_nccl: bool = False use_nccl: bool = False
# Whether to enablea HIP
use_hip: bool = False
# Whether to enable RCCL
use_rccl: bool = False
# Whether to enable HDFS # Whether to enable HDFS
use_hdfs: bool = False use_hdfs: bool = False
# Whether to enable Azure Storage # Whether to enable Azure Storage