diff --git a/src/common/survival_util.h b/src/common/survival_util.h index e891edb54..c5f134fc1 100644 --- a/src/common/survival_util.h +++ b/src/common/survival_util.h @@ -25,12 +25,12 @@ DECLARE_FIELD_ENUM_CLASS(xgboost::common::ProbabilityDistributionType); namespace xgboost { namespace common { -#ifndef __CUDACC__ +#if !defined(__CUDACC__) && !defined(__HIP_PLATFORM_AMD__) using std::log; using std::fmax; -#endif // __CUDACC__ +#endif // __CUDACC__ && __HIP_PLATFORM_AMD__ enum class CensoringType : uint8_t { kUncensored, kRightCensored, kLeftCensored, kIntervalCensored diff --git a/src/metric/metric.cc b/src/metric/metric.cc index ebb579827..2b805185c 100644 --- a/src/metric/metric.cc +++ b/src/metric/metric.cc @@ -84,7 +84,7 @@ DMLC_REGISTRY_LINK_TAG(elementwise_metric); DMLC_REGISTRY_LINK_TAG(multiclass_metric); DMLC_REGISTRY_LINK_TAG(survival_metric); DMLC_REGISTRY_LINK_TAG(rank_metric); -#ifdef XGBOOST_USE_CUDA +#if defined(XGBOOST_USE_CUDA) || defined(XGBOOST_USE_HIP) DMLC_REGISTRY_LINK_TAG(auc_gpu); DMLC_REGISTRY_LINK_TAG(rank_metric_gpu); #endif