Improved multi-node multi-GPU random forests. (#4238)

* Improved multi-node multi-GPU random forests.

- removed rabit::Broadcast() from each invocation of column sampling
- instead, syncing the PRNG seed when a ColumnSampler() object is constructed
- this makes non-trivial column sampling significantly faster in the distributed case
- refactored distributed GPU tests
- added distributed random forests tests
This commit is contained in:
Andy Adinets
2019-03-13 00:36:28 +01:00
committed by Rory Mitchell
parent 99a714be64
commit b833b642ec
9 changed files with 162 additions and 216 deletions

View File

@@ -7,5 +7,5 @@ cd ..
pytest -v -s --fulltrace -m "(not slow) and mgpu" tests/python-gpu
./testxgboost --gtest_filter=*.MGPU_*
cd tests/distributed-gpu
./runtests-gpu.sh
cd tests/distributed
./runtests-gpu.sh