Function MakeLabels

tests/cpp/helpers.cc:400–421

namespace {
void MakeLabels(DeviceOrd device, bst_idx_t n_samples, bst_target_t n_classes,
                bst_target_t n_targets, std::shared_ptr<DMatrix> out) {
  RandomDataGenerator gen{n_samples, n_targets, 0.0f};
  if (n_classes != 0) {
    gen.Lower(0).Upper(n_classes).GenerateDense(out->Info().labels.Data());
    out->Info().labels.Reshape(n_samples, n_targets);
    auto& h_labels = out->Info().labels.Data()->HostVector();
    for (auto& v : h_labels) {
      v = static_cast<float>(static_cast<uint32_t>(v));
    }
  } else {
    gen.GenerateDense(out->Info().labels.Data());
    CHECK_EQ(out->Info().labels.Size(), n_samples * n_targets);
    out->Info().labels.Reshape(n_samples, n_targets);
  }
  if (device.IsCUDA()) {
    out->Info().labels.Data()->SetDevice(device);
    out->Info().labels.Data()->ConstDevicePointer();
    out->Info().feature_types.SetDevice(device);
    out->Info().feature_types.ConstDevicePointer();
  }
}

[[nodiscard]] bool DecompAllowFallback() {
#if defined(XGBOOST_USE_NVCOMP)