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

move nthread to local var

Browse Source
This commit is contained in:
tqchen 2014-09-02 09:06:24 -07:00
parent 28128a1b6e
commit 70219ee1ae
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/tree/updater_refresh-inl.hpp
View File

@ -33,6 +33,7 @@ class TreeRefresher: public IUpdater {
std::vector< std::vector<TStats> > stemp; std::vector< std::vector<TStats> > stemp;
std::vector<RegTree::FVec> fvec_temp; std::vector<RegTree::FVec> fvec_temp;
// setup temp space for each thread // setup temp space for each thread
int nthread;
#pragma omp parallel #pragma omp parallel
{ {
nthread = omp_get_num_threads(); nthread = omp_get_num_threads();
@ -126,8 +127,6 @@ class TreeRefresher: public IUpdater {
this->Refresh(gstats, tree[nid].cright(), p_tree); this->Refresh(gstats, tree[nid].cright(), p_tree);
} }
} }
// number of thread in the data
int nthread;
// training parameter // training parameter
TrainParam param; TrainParam param;
}; };