Copy output data for argsort. (#6866)

Fix GPU AUC.
2021-04-16 21:05:01 +08:00
parent a5d7094a45
commit 1b26a2a561
2 changed files with 17 additions and 8 deletions
--- a/src/metric/auc.cu
+++ b/src/metric/auc.cu
@@ -269,7 +269,7 @@ float GPUMultiClassAUCOVR(common::Span<float const> predts, MetaInfo const &info
      });

  // unique values are sparse, so we need a CSR style indptr
-  dh::TemporaryArray<uint32_t> unique_class_ptr(class_ptr.size() + 1);
+  dh::TemporaryArray<uint32_t> unique_class_ptr(class_ptr.size());
  auto d_unique_class_ptr = dh::ToSpan(unique_class_ptr);
  auto n_uniques = dh::SegmentedUniqueByKey(
      thrust::cuda::par(alloc),