29 lines
743 B
Plaintext
29 lines
743 B
Plaintext
/**
|
|
* 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
|