[HOTFIX] distributed training with hist method (#4716)
* add parallel test for hist.EvalualiteSplit * update test_openmp.py * update test_openmp.py * update test_openmp.py * update test_openmp.py * update test_openmp.py * fix OMP schedule policy * fix clang-tidy * add logging: total_num_bins * fix * fix * test * replace guided OPENMP policy with static in updater_quantile_hist.cc
This commit is contained in:
committed by
Philip Hyunsu Cho
parent
c0ffe65f5c
commit
ef9af33a00
@@ -225,6 +225,14 @@ class QuantileHistMock : public QuantileHistMaker {
|
||||
|
||||
delete dmat;
|
||||
}
|
||||
|
||||
void TestEvaluateSplitParallel(const GHistIndexBlockMatrix &quantile_index_block,
|
||||
const RegTree &tree) {
|
||||
omp_set_num_threads(2);
|
||||
TestEvaluateSplit(quantile_index_block, tree);
|
||||
omp_set_num_threads(1);
|
||||
}
|
||||
|
||||
};
|
||||
|
||||
int static constexpr kNRows = 8, kNCols = 16;
|
||||
|
||||
Reference in New Issue
Block a user