6deaec8027
- Pass obj info into tree updater as const pointer. This way we don't have to initialize the learner model param before configuring gbm, hence breaking up the dependency of configurations.
28 lines
852 B
C++
28 lines
852 B
C++
/**
|
|
* Copyright 2023 by XGBoost contributors
|
|
*/
|
|
#include <gtest/gtest.h>
|
|
#include <xgboost/context.h> // for Context
|
|
#include <xgboost/task.h> // for ObjInfo
|
|
#include <xgboost/tree_updater.h> // for TreeUpdater
|
|
|
|
#include <memory> // for unique_ptr
|
|
|
|
namespace xgboost {
|
|
TEST(Updater, HasNodePosition) {
|
|
Context ctx;
|
|
ObjInfo task{ObjInfo::kRegression, true, true};
|
|
std::unique_ptr<TreeUpdater> up{TreeUpdater::Create("grow_histmaker", &ctx, &task)};
|
|
ASSERT_TRUE(up->HasNodePosition());
|
|
|
|
up.reset(TreeUpdater::Create("grow_quantile_histmaker", &ctx, &task));
|
|
ASSERT_TRUE(up->HasNodePosition());
|
|
|
|
#if defined(XGBOOST_USE_CUDA)
|
|
ctx.gpu_id = 0;
|
|
up.reset(TreeUpdater::Create("grow_gpu_hist", &ctx, &task));
|
|
ASSERT_TRUE(up->HasNodePosition());
|
|
#endif // defined(XGBOOST_USE_CUDA)
|
|
}
|
|
} // namespace xgboost
|