Remove hack.

dmlc · Sep 8, 2022 · cc90c2b · cc90c2b
1 parent 7654b88
commit cc90c2b
Showing 1 changed file with 0 additions and 9 deletions.
diff --git a/src/data/iterative_dmatrix.cc b/src/data/iterative_dmatrix.cc
@@ -33,15 +33,6 @@ IterativeDMatrix::IterativeDMatrix(DataIterHandle iter_handle, DMatrixHandle pro
     CHECK_EQ(d, batch_param_.gpu_id) << msg;
   }
 
-  int32_t max_device{d};
-  rabit::Allreduce<rabit::op::Max>(&max_device, 1);
-  if (max_device != d) {
-    CHECK_EQ(MakeProxy(proxy_)->Info().num_row_, 0)
-        << "max device:" << max_device << " device:" << d;
-    CHECK_NE(d, Context::kCpuId) << msg;
-    d = max_device;
-  }
-
   batch_param_ = BatchParam{d, max_bin};
   batch_param_.sparse_thresh = 0.2;  // default from TrainParam