Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix timer annotation. (#9057)

Browse Source
This commit is contained in:
Jiaming Yuan 2023-04-21 22:53:58 +08:00 committed by GitHub
parent 3b742dc4f1
commit 7032981350
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/tree/updater_quantile_hist.cc
View File

@ -294,6 +294,7 @@ class MultiTargetHistBuilder {
std::vector<bst_node_t> *p_out_position) { std::vector<bst_node_t> *p_out_position) {
monitor_->Start(__func__); monitor_->Start(__func__);
if (!task_->UpdateTreeLeaf()) { if (!task_->UpdateTreeLeaf()) {
monitor_->Stop(__func__);
return; return;
} }
for (auto const &part : partitioner_) { for (auto const &part : partitioner_) {
@ -397,6 +398,7 @@ class HistBuilder {
evaluator_ = std::make_unique<HistEvaluator<CPUExpandEntry>>(ctx_, this->param_, fmat->Info(), evaluator_ = std::make_unique<HistEvaluator<CPUExpandEntry>>(ctx_, this->param_, fmat->Info(),
col_sampler_); col_sampler_);
p_last_tree_ = p_tree; p_last_tree_ = p_tree;
monitor_->Stop(__func__);
} }
void EvaluateSplits(DMatrix *p_fmat, RegTree const *p_tree, void EvaluateSplits(DMatrix *p_fmat, RegTree const *p_tree,