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

Performance optimizations for Intel CPUs (#3957)

Browse Source 
* Initial performance optimizations for xgboost

* remove includes

* revert float->double

* fix for CI

* fix for CI

* fix for CI

* fix for CI

* fix for CI

* fix for CI

* fix for CI

* fix for CI

* fix for CI

* fix for CI

* Check existence of _mm_prefetch and __builtin_prefetch

* Fix lint
This commit is contained in:
Egor Smirnov
2019-01-09 08:08:13 +03:00
committed by Philip Hyunsu Cho
parent dade7c3aff
commit 5f151c5cf3
7 changed files with 145 additions and 43 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/common/hist_util.h
View File

@@ -238,6 +238,7 @@ class GHistBuilder {
inline void Init(size_t nthread, uint32_t nbins) {
nthread_ = nthread;
nbins_ = nbins;
thread_init_.resize(nthread_);
}
// construct a histogram via histogram aggregation
@@ -259,6 +260,7 @@ class GHistBuilder {
/*! \brief number of all bins over all features */
uint32_t nbins_;
std::vector<GHistEntry> data_;
std::vector<size_t> thread_init_;
};