Use nccl group calls to prevent from dead lock. (#4113)

* launch all reduce sequentially.
* Fix gpu_exact test memory leak.
This commit is contained in:
Jiaming Yuan
2019-02-08 06:12:39 +08:00
committed by GitHub
parent 05243642bb
commit f8ca2960fc
4 changed files with 18 additions and 15 deletions

View File

@@ -42,6 +42,9 @@ TEST(GPUExact, Update) {
ASSERT_NEAR(tree.Stat(2).sum_hess, 2, kRtEps);
ASSERT_NEAR(tree.Stat(0).loss_chg, 0.8f, kRtEps);
delete dmat;
delete p_gpuexact_maker;
}
} // namespace tree