diff --git a/src/common/ranking_utils.cc b/src/common/ranking_utils.cc index d831b551c..cc73d15c2 100644 --- a/src/common/ranking_utils.cc +++ b/src/common/ranking_utils.cc @@ -62,7 +62,7 @@ common::Span RankingCache::MakeRankOnCPU(Context const* ctx, return rank; } -#if !defined(XGBOOST_USE_CUDA) +#if !defined(XGBOOST_USE_CUDA) && !defined(XGBOOST_USE_HIP) void RankingCache::InitOnCUDA(Context const*, MetaInfo const&) { common::AssertGPUSupport(); } common::Span RankingCache::MakeRankOnCUDA(Context const*, common::Span) { @@ -108,9 +108,9 @@ void NDCGCache::InitOnCPU(Context const* ctx, MetaInfo const& info) { }); } -#if !defined(XGBOOST_USE_CUDA) +#if !defined(XGBOOST_USE_CUDA) && !defined(XGBOOST_USE_HIP) void NDCGCache::InitOnCUDA(Context const*, MetaInfo const&) { common::AssertGPUSupport(); } -#endif // !defined(XGBOOST_USE_CUDA) +#endif // !defined(XGBOOST_USE_CUDA) && !defined(XGBOOST_USE_HIP) DMLC_REGISTER_PARAMETER(LambdaRankParam); @@ -119,9 +119,9 @@ void MAPCache::InitOnCPU(Context const*, MetaInfo const& info) { CheckMapLabels(h_label, [](auto beg, auto end, auto op) { return std::all_of(beg, end, op); }); } -#if !defined(XGBOOST_USE_CUDA) +#if !defined(XGBOOST_USE_CUDA) && !defined(XGBOOST_USE_HIP) void MAPCache::InitOnCUDA(Context const*, MetaInfo const&) { common::AssertGPUSupport(); } -#endif // !defined(XGBOOST_USE_CUDA) +#endif // !defined(XGBOOST_USE_CUDA) && !defined(XGBOOST_USE_HIP) std::string ParseMetricName(StringView name, StringView param, position_t* topn, bool* minus) { std::string out_name; diff --git a/src/metric/rank_metric.h b/src/metric/rank_metric.h index b3b121973..ca6b8b61d 100644 --- a/src/metric/rank_metric.h +++ b/src/metric/rank_metric.h @@ -23,7 +23,7 @@ PackedReduceResult MAPScore(Context const *ctx, MetaInfo const &info, HostDeviceVector const &predt, bool minus, std::shared_ptr p_cache); -#if !defined(XGBOOST_USE_CUDA) +#if !defined(XGBOOST_USE_CUDA) && !defined(XGBOOST_USE_HIP) inline PackedReduceResult NDCGScore(Context const *, MetaInfo const &, HostDeviceVector const &, bool, std::shared_ptr) {