Disable HT for DMatrix creation (#6386)

Co-authored-by: SHVETS, KIRILL <kirill.shvets@intel.com>
This commit is contained in:
ShvetsKS
2020-11-14 17:18:33 +03:00
committed by GitHub
parent fcd6fad822
commit 512b464cfa
3 changed files with 12 additions and 2 deletions

View File

@@ -844,7 +844,7 @@ void SparsePage::Push(const SparsePage &batch) {
template <typename AdapterBatchT>
uint64_t SparsePage::Push(const AdapterBatchT& batch, float missing, int nthread) {
// Set number of threads but keep old value so we can reset it after
int nthread_original = common::OmpSetNumThreads(&nthread);
int nthread_original = common::OmpSetNumThreadsWithoutHT(&nthread);
auto& offset_vec = offset.HostVector();
auto& data_vec = data.HostVector();