Merge branch 'master' into sync-condition-2023Oct11
This commit is contained in:
28
tests/cpp/tree/gpu_hist/test_expand_entry.cu
Normal file
28
tests/cpp/tree/gpu_hist/test_expand_entry.cu
Normal 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
|
||||
Reference in New Issue
Block a user