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
|
/**
* Copyright 2023, XGBoost Contributors
*/
#include <gtest/gtest.h>
#include <xgboost/json.h>
#include <xgboost/tree_model.h> // for RegTree
#include "../../../../src/tree/gpu_hist/expand_entry.cuh"
namespace xgboost::tree {
TEST(ExpandEntry, IOGPU) {
DeviceSplitCandidate split;
GPUExpandEntry entry{RegTree::kRoot, 0, split, 3.0, 1.0, 2.0};
Json je{Object{}};
entry.Save(&je);
GPUExpandEntry loaded;
loaded.Load(je);
ASSERT_EQ(entry.base_weight, loaded.base_weight);
ASSERT_EQ(entry.left_weight, loaded.left_weight);
ASSERT_EQ(entry.right_weight, loaded.right_weight);
ASSERT_EQ(entry.GetDepth(), loaded.GetDepth());
ASSERT_EQ(entry.GetLossChange(), loaded.GetLossChange());
}
} // namespace xgboost::tree
|
