From cc90c2bc16db715e73b90874ece4b53527e41423 Mon Sep 17 00:00:00 2001
From: fis <jm.yuan@outlook.com>
Date: Thu, 8 Sep 2022 18:58:12 +0800
Subject: [PATCH] Remove hack.

---
 src/data/iterative_dmatrix.cc | 9 ---------
 1 file changed, 9 deletions(-)

diff --git a/src/data/iterative_dmatrix.cc b/src/data/iterative_dmatrix.cc
index c8752785f43b..60591a1d91b8 100644
--- 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