Old CUDA.

dmlc · Oct 9, 2021 · 1b5808c · 1b5808c
1 parent 8b5a15a
commit 1b5808c
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 5 deletions.
diff --git a/src/metric/auc.cc b/src/metric/auc.cc
@@ -461,8 +461,7 @@ class EvalAUCPR : public EvalAUC<EvalAUCPR> {
     auto n_threads = tparam_->Threads();
     if (tparam_->gpu_id == GenericParameter::kCpuId) {
       auto labels = info.labels_.ConstHostSpan();
-      if (std::any_of(labels.cbegin(), labels.cend(),
-                      [](float y) { return y < 0.0f || y > 1.0f; })) {
+      if (std::any_of(labels.cbegin(), labels.cend(), PRAUCLabelInvalid{})) {
         InvalidLabels();
       }
       std::tie(auc, valid_groups) =

diff --git a/src/metric/auc.cu b/src/metric/auc.cu
@@ -897,9 +897,8 @@ GPURankingPRAUC(common::Span<float const> predts, MetaInfo const &info,
 
   dh::XGBDeviceAllocator<char> alloc;
   auto labels = info.labels_.ConstDeviceSpan();
-  if (thrust::any_of(
-          thrust::cuda::par(alloc), dh::tbegin(labels), dh::tend(labels),
-          [] XGBOOST_DEVICE(float y) { return y < 0.0f || y > 1.0f; })) {
+  if (thrust::any_of(thrust::cuda::par(alloc), dh::tbegin(labels),
+                     dh::tend(labels), PRAUCLabelInvalid{})) {
     InvalidLabels();
   }
   auto d_weights = info.weights_.ConstDeviceSpan();

diff --git a/src/metric/auc.h b/src/metric/auc.h
@@ -105,6 +105,10 @@ inline void InvalidGroupAUC() {
             << "least 2 pairs of samples.";
 }
 
+struct PRAUCLabelInvalid {
+  XGBOOST_DEVICE bool operator()(float y) { return y < 0.0f || y > 1.0f; }
+};
+
 inline void InvalidLabels() {
   LOG(FATAL) << "PR-AUC supports only binary relevance for learning to rank.";
 }