From d0774a78e4041d67731ab65a3fb9f6d86a3fc8b2 Mon Sep 17 00:00:00 2001 From: Hui Liu <96135754+amdsc21@users.noreply.github.com> Date: Mon, 30 Oct 2023 12:01:24 -0700 Subject: [PATCH] add hip to config --- R-package/src/xgboost_custom.cc | 2 +- python-package/packager/build_config.py | 4 ++++ 2 files changed, 5 insertions(+), 1 deletion(-) diff --git a/R-package/src/xgboost_custom.cc b/R-package/src/xgboost_custom.cc index f196297ec..92f8a8e1f 100644 --- a/R-package/src/xgboost_custom.cc +++ b/R-package/src/xgboost_custom.cc @@ -32,7 +32,7 @@ namespace common { bool CheckNAN(double v) { return ISNAN(v); } -#if !defined(XGBOOST_USE_CUDA) +#if !defined(XGBOOST_USE_CUDA) && !defined(XGBOOST_USE_HIP) double LogGamma(double v) { return lgammafn(v); } diff --git a/python-package/packager/build_config.py b/python-package/packager/build_config.py index 26392a897..517dc17f0 100644 --- a/python-package/packager/build_config.py +++ b/python-package/packager/build_config.py @@ -15,6 +15,10 @@ class BuildConfiguration: # pylint: disable=R0902 use_cuda: bool = False # Whether to enable NCCL use_nccl: bool = False + # Whether to enablea HIP + use_hip: bool = False + # Whether to enable RCCL + use_rccl: bool = False # Whether to enable HDFS use_hdfs: bool = False # Whether to enable Azure Storage