move processor shutdown

ZiyueXu77 · May 10, 2024 · 20bb965 · 20bb965
1 parent f3967c5
commit 20bb965
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/src/c_api/c_api.cc b/src/c_api/c_api.cc
@@ -45,7 +45,6 @@
 #include "xgboost/span.h"                    // for Span
 #include "xgboost/string_view.h"             // for StringView, operator<<
 #include "xgboost/version_config.h"          // for XGBOOST_VER_MAJOR, XGBOOST_VER_MINOR, XGBOOS...
-#include "../processing/processor.h"  // for Processor
 
 #if defined(XGBOOST_USE_FEDERATED)
 #include "../../plugin/federated/federated_server.h"
@@ -1738,8 +1737,6 @@ XGB_DLL int XGCommunicatorInit(char const* json_config) {
 XGB_DLL int XGCommunicatorFinalize() {
   API_BEGIN();
   collective::Finalize();
-  processor_instance->Shutdown();
-  processor_instance = nullptr;
   API_END();
 }
 

diff --git a/src/collective/communicator.cc b/src/collective/communicator.cc
@@ -100,6 +100,10 @@ void Communicator::Init(Json const& config) {
 void Communicator::Finalize() {
   communicator_->Shutdown();
   communicator_.reset(new NoOpCommunicator());
+  if (processor_instance != nullptr) {
+    processor_instance->Shutdown();
+    processor_instance = nullptr;
+  }
 }
 #endif
 }  // namespace xgboost::collective