switch to allgatherV for encrypted message with varying lenghts

ZiyueXu77 · Apr 15, 2024 · a3ddf7d · a3ddf7d
1 parent 30b7ed5
commit a3ddf7d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/tree/hist/histogram.h b/src/tree/hist/histogram.h
@@ -238,7 +238,7 @@ class HistogramBuilder {
       // Perform AllGather
       auto hist_vec = std::vector<std::int8_t>(hist_data.data(),
                                                hist_data.data() + hist_data.size());
-      auto hist_entries = collective::Allgather(hist_vec);
+      auto hist_entries = collective::AllgatherV(hist_vec);
       // Call interface here to post-process the messages
       std::vector<double> hist_aggr = processor_instance->HandleAggregation(hist_entries.data(), hist_entries.size());