Refactor out row partitioning logic from gpu_hist, introduce caching device vectors (#4554)

2019-06-20 18:24:09 +12:00
parent 0c50f8417a
commit 221e163185
7 changed files with 582 additions and 345 deletions
--- a/tests/cpp/common/test_device_helpers.cu
+++ b/tests/cpp/common/test_device_helpers.cu
@@ -97,7 +97,8 @@ TEST(bulkAllocator, Test) {
 }

 // Test thread safe max reduction
-TEST(AllReducer, HostMaxAllReduce) {
+#if defined(XGBOOST_USE_NCCL)
+TEST(AllReducer, MGPU_HostMaxAllReduce) {
  dh::AllReducer reducer;
  size_t num_threads = 50;
  std::vector<std::vector<size_t>> thread_data(num_threads);
@@ -112,3 +113,4 @@ TEST(AllReducer, HostMaxAllReduce) {
    ASSERT_EQ(data.front(), num_threads - 1);
  }
 }
+#endif