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
|
