wrap plugin calls into federated

ZiyueXu77 · Aug 1, 2024 · 7480ed3 · 7480ed3
1 parent aa5b51b
commit 7480ed3
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/tree/gpu_hist/histogram.cu b/src/tree/gpu_hist/histogram.cu
@@ -395,6 +395,7 @@ void DeviceHistogramBuilder::BuildHistogram(Context const* ctx,
     this->p_impl_->BuildHistogram(ctx->CUDACtx(), matrix, feature_groups,
                                   gpair, ridx, histogram, rounding);
   } else {
+  #if defined(XGBOOST_USE_FEDERATED)
     // Encrypted vertical, build histogram using federated plugin
     auto const &comm = collective::GlobalCommGroup()->Ctx(ctx, DeviceOrd::CPU());
     auto const &fed = dynamic_cast<collective::FederatedComm const &>(comm);
@@ -480,6 +481,9 @@ void DeviceHistogramBuilder::BuildHistogram(Context const* ctx,
     dh::safe_cuda(cudaMemcpyAsync(histogram.data(), host_histogram.data(),
                                   histogram.size() * sizeof(GradientPairInt64),
                                   cudaMemcpyHostToDevice));
+  #else
+    LOG(FATAL) << error::NoFederated();
+  #endif
   }
 }
 }  // namespace xgboost::tree