add include conditions

ZiyueXu77 · Jul 26, 2024 · 7df5955 · 7df5955
1 parent 32c3014
commit 7df5955
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/src/tree/updater_gpu_hist.cu b/src/tree/updater_gpu_hist.cu
@@ -582,6 +582,8 @@ struct GPUHistMakerDevice {
 
     monitor.Stop(__func__);
   }
+#else
+  LOG(FATAL) << error::NoFederated();
 #endif
 
   /**
@@ -622,6 +624,8 @@ struct GPUHistMakerDevice {
     if (collective::IsDistributed() && info_.IsRowSplit() && collective::IsEncrypted()) {
       #if defined(XGBOOST_USE_FEDERATED)
         this->AllReduceHistEncrypted(hist_nidx.at(0), hist_nidx.size());
+      #else
+        LOG(FATAL) << error::NoFederated();
       #endif
     } else {
       this->AllReduceHist(hist_nidx.at(0), hist_nidx.size());
@@ -638,6 +642,8 @@ struct GPUHistMakerDevice {
         if (collective::IsDistributed() && info_.IsRowSplit() && collective::IsEncrypted()) {
           #if defined(XGBOOST_USE_FEDERATED)
             this->AllReduceHistEncrypted(subtraction_trick_nidx, 1);
+          #else
+            LOG(FATAL) << error::NoFederated();
           #endif
         } else {
           this->AllReduceHist(subtraction_trick_nidx, 1);
@@ -718,6 +724,8 @@ struct GPUHistMakerDevice {
     if (collective::IsDistributed() && info_.IsRowSplit() && collective::IsEncrypted()) {
       #if defined(XGBOOST_USE_FEDERATED)
         this->AllReduceHistEncrypted(kRootNIdx, 1);
+      #else
+        LOG(FATAL) << error::NoFederated();
       #endif
     } else {
       this->AllReduceHist(kRootNIdx, 1);