rebase.

dmlc · Sep 18, 2023 · 1b6e762 · 1b6e762
1 parent ae18374
commit 1b6e762
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/src/predictor/gpu_predictor.cu b/src/predictor/gpu_predictor.cu
@@ -823,7 +823,6 @@ class ColumnSplitHelper {
  ctx_->Ordinal(), decision_storage->data().get(), decision_storage->size());
  collective::AllReduce<collective::Operation::kBitwiseAND>(
  ctx_->Ordinal(), missing_storage->data().get(), missing_storage->size());
- collective::Synchronize(ctx_->Ordinal());
  }
 
  void ResizeBitVectors(dh::caching_device_vector<BitType>* decision_storage,

diff --git a/tests/cpp/tree/test_gpu_hist.cu b/tests/cpp/tree/test_gpu_hist.cu
@@ -486,7 +486,7 @@ RegTree GetApproxTree(Context const* ctx, DMatrix* dmat) {
  TrainParam param;
  param.UpdateAllowUnknown(Args{});
 
- linalg::Matrix<GradientPair> gpair({dmat->Info().num_row_}, ctx->Ordinal());
+ linalg::Matrix<GradientPair> gpair({dmat->Info().num_row_}, ctx->Device());
  gpair.Data()->Copy(GenerateRandomGradients(dmat->Info().num_row_));
 
  std::vector<HostDeviceVector<bst_node_t>> position(1);