AddPlan (#5450)

Co-authored-by: oneflow-ci-bot <69100618+oneflow-ci-bot@users.noreply.github.com>

oneflow/core/job/runtime.cpp

@@ -128,7 +128,8 @@ void Runtime::NewAllGlobal(const Plan& plan, size_t total_piece_num, bool is_exp
   Global<MemoryAllocator>::New();
   Global<RegstMgr>::New(plan);
   Global<ActorMsgBus>::New();
-  Global<ThreadMgr>::New(plan);
+  Global<ThreadMgr>::New();
+  Global<ThreadMgr>::Get()->AddPlan(plan);
   Global<boxing::collective::CollectiveBoxingDeviceCtxPoller>::New();
   Global<RuntimeJobDescs>::New(plan.job_confs().job_id2job_conf());
   Global<summary::EventsWriter>::New();

oneflow/core/thread/thread_manager.cpp

@@ -42,7 +42,7 @@ Thread* ThreadMgr::GetThrd(int64_t thrd_id) {
   return iter->second.get();
 }
 
-ThreadMgr::ThreadMgr(const Plan& plan) {
+void ThreadMgr::AddPlan(const Plan& plan) {
   const int64_t this_rank = GlobalProcessCtx::Rank();
   for (const TaskProto& task : plan.task()) {
     TaskId task_id = DeserializeTaskIdFromInt64(task.task_id());

oneflow/core/thread/thread_manager.h

@@ -29,14 +29,14 @@ class Plan;
 class ThreadMgr final {
  public:
   OF_DISALLOW_COPY_AND_MOVE(ThreadMgr);
-  ThreadMgr() = delete;
+  ThreadMgr() = default;
   ~ThreadMgr();
 
+  void AddPlan(const Plan& plan);
   Thread* GetThrd(int64_t thrd_id);
 
  private:
   friend class Global<ThreadMgr>;
-  explicit ThreadMgr(const Plan& plan);
 
   HashMap<int64_t, std::unique_ptr<Thread>> threads_;
 };