xgboost_lib-sys 3.0.5

Native bindings to the xgboost library
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45

/**
 * Copyright 2020-2025, XGBoost Contributors
 */
#include <gtest/gtest.h>
#include <xgboost/c_api.h>
#include <xgboost/global_config.h>
#include <xgboost/json.h>
#include <xgboost/logging.h>

namespace xgboost {
TEST(GlobalConfiguration, Verbosity) {
  // Configure verbosity via global configuration
  Json config{JsonObject()};
  config["verbosity"] = String("0");
  auto& global_config = *GlobalConfigThreadLocalStore::Get();
  FromJson(config, &global_config);
  // Now verbosity should be updated
  EXPECT_EQ(ConsoleLogger::GlobalVerbosity(), ConsoleLogger::LogVerbosity::kSilent);
  EXPECT_NE(ConsoleLogger::LogVerbosity::kSilent, ConsoleLogger::DefaultVerbosity());
  // GetConfig() should also return updated verbosity
  Json current_config{ToJson(*GlobalConfigThreadLocalStore::Get())};
  EXPECT_EQ(get<String>(current_config["verbosity"]), "0");
}

TEST(GlobalConfiguration, UseRMM) {
  Json config{JsonObject()};
  config["use_rmm"] = String("true");
  auto& global_config = *GlobalConfigThreadLocalStore::Get();
  FromJson(config, &global_config);
  // GetConfig() should return updated use_rmm flag
  Json current_config{ToJson(*GlobalConfigThreadLocalStore::Get())};
  EXPECT_EQ(get<String>(current_config["use_rmm"]), "1");
}

TEST(GlobalConfiguration, Threads) {
  char const* config;
  ASSERT_EQ(XGBGetGlobalConfig(&config), 0);
  auto jconfig = Json::Load(config);
  auto nthread = get<Integer const>(jconfig["nthread"]);
  ASSERT_LE(nthread, 0);
  auto n_omp = omp_get_num_threads();
  ASSERT_EQ(XGBSetGlobalConfig(config), 0);
  ASSERT_EQ(n_omp, omp_get_num_threads());
}
}  // namespace xgboost