integration with interface initial attempt

ZiyueXu77 · Mar 22, 2024 · 406cda3 · 406cda3
1 parent da0f7a6
commit 406cda3
Show file tree

Hide file tree

Showing 4 changed files with 11 additions and 23 deletions.
diff --git a/src/collective/aggregator.h b/src/collective/aggregator.h
@@ -101,14 +101,6 @@ void ApplyWithLabels(Context const*, MetaInfo const& info, HostDeviceVector<T>*
       // Under secure mode, gpairs will be processed to vector and encrypt
       // information only available on rank 0
 
-      xgboost::processing::ProcessorLoader loader;
-      auto processor = loader.load("dummy");
-      if (collective::GetRank() == 0) {
-        processor->Initialize(true, {});
-      } else {
-        processor->Initialize(false, {});
-      }
-
       std::size_t buffer_size{};
       std::int8_t *buffer;
       //common::Span<std::int8_t> buffer;
@@ -128,7 +120,7 @@ void ApplyWithLabels(Context const*, MetaInfo const& info, HostDeviceVector<T>*
           std::cout << "-----------Call Interface for gp encryption and broadcast"
           << ", size of gpairs: " << vector_gh.size()
           << " ----------------------" << std::endl;
-          auto buf = processor->ProcessGHPairs(vector_gh);
+          auto buf = processor_instance->ProcessGHPairs(vector_gh);
           buffer_size = buf.size();
           buffer = reinterpret_cast<std::int8_t *>(buf.data());
           std::cout << "buffer size: " << buffer_size << std::endl;
@@ -146,7 +138,7 @@ void ApplyWithLabels(Context const*, MetaInfo const& info, HostDeviceVector<T>*
 
       // call HandleGHPairs
       xgboost::common::Span<int8_t> buf = xgboost::common::Span<int8_t>(buffer, buffer_size);
-      processor->HandleGHPairs(buf);
+      processor_instance->HandleGHPairs(buf);
 
 
 

diff --git a/src/learner.cc b/src/learner.cc
@@ -68,7 +68,7 @@ const char* kMaxDeltaStepDefaultValue = "0.7";
 }  // anonymous namespace
 
 DECLARE_FIELD_ENUM_CLASS(xgboost::MultiStrategy);
-
+xgboost::processing::Processor *processor_instance;
 namespace xgboost {
 Learner::~Learner() = default;
 namespace {
@@ -497,11 +497,11 @@ class LearnerConfiguration : public Learner {
       std::cout << "configure interface here???????????????" << std::endl;
     }
       xgboost::processing::ProcessorLoader loader;
-      auto processor = loader.load("dummy");
+      processor_instance = loader.load("dummy");
       if (collective::GetRank() == 0) {
-        processor->Initialize(true, {});
+          processor_instance->Initialize(true, {});
       } else {
-        processor->Initialize(false, {});
+          processor_instance->Initialize(false, {});
       }
 
 

diff --git a/src/processing/processor.h b/src/processing/processor.h
@@ -105,4 +105,7 @@ class ProcessorLoader {
 
     void unload();
 };
+
 }  // namespace xgboost::processing
+
+extern xgboost::processing::Processor *processor_instance;
diff --git a/src/tree/hist/histogram.h b/src/tree/hist/histogram.h
@@ -120,15 +120,8 @@ class HistogramBuilder {
       if ((collective::GetRank() == 0)) {
         std::cout << "------------CALL interface to transmit row & gidx---------" << std::endl;
       }
-      xgboost::processing::ProcessorLoader loader;
-      auto processor = loader.load("dummy");
-      if (collective::GetRank() == 0) {
-        processor->Initialize(true, {});
-      } else {
-        processor->Initialize(false, {});
-      }
-      processor->InitAggregationContext(gidx);
-      processor->ProcessAggregation(nodes_to_build, row_set_collection);
+      processor_instance->InitAggregationContext(gidx);
+      processor_instance->ProcessAggregation(nodes_to_build, row_set_collection);
     } else {
       // Parallel processing by nodes and data in each node
       common::ParallelFor2d(space, this->n_threads_, [&](size_t nid_in_set, common::Range1d r) {