Merge branch 'master' into sync-condition-2023Oct11

This commit is contained in:
Hui Liu
2023-10-24 10:46:38 -07:00
14 changed files with 336 additions and 76 deletions

View File

@@ -0,0 +1,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