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

added tracking execution time for UpdatePredictionCache function (#5107)

Browse Source
This commit is contained in:
Egor Smirnov 2019-12-09 20:32:56 +03:00 committed by Jiaming Yuan
parent 38763aa4fa
commit b1789b0346
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/tree/updater_quantile_hist.cc
View File

@ -351,13 +351,14 @@ void QuantileHistMaker::Builder::Update(const GHistIndexMatrix& gmat,
bool QuantileHistMaker::Builder::UpdatePredictionCache( bool QuantileHistMaker::Builder::UpdatePredictionCache(
const DMatrix* data, const DMatrix* data,
HostDeviceVector<bst_float>* p_out_preds) { HostDeviceVector<bst_float>* p_out_preds) {
std::vector<bst_float>& out_preds = p_out_preds->HostVector();
// p_last_fmat_ is a valid pointer as long as UpdatePredictionCache() is called in // p_last_fmat_ is a valid pointer as long as UpdatePredictionCache() is called in
// conjunction with Update(). // conjunction with Update().
if (!p_last_fmat_ || !p_last_tree_ || data != p_last_fmat_) { if (!p_last_fmat_ || !p_last_tree_ || data != p_last_fmat_) {
return false; return false;
} }
builder_monitor_.Start("UpdatePredictionCache");
std::vector<bst_float>& out_preds = p_out_preds->HostVector();
if (leaf_value_cache_.empty()) { if (leaf_value_cache_.empty()) {
leaf_value_cache_.resize(p_last_tree_->param.num_nodes, leaf_value_cache_.resize(p_last_tree_->param.num_nodes,
@ -385,6 +386,8 @@ bool QuantileHistMaker::Builder::UpdatePredictionCache(
} }
} }
} }
builder_monitor_.Stop("UpdatePredictionCache");
return true; return true;
} }