Method SetIdState

oneflow/core/framework/multi_client_session_context.cpp:221–228

}

void MultiClientSessionContext::SetIdState(const IdState& id_state) {
  CHECK(Singleton<IDMgr>::Get() != nullptr);
  CHECK(Singleton<TaskStreamIndexManager>::Get() != nullptr);
  CHECK(Singleton<LazyJobBuildAndInferCtxMgr>::Get() != nullptr);
  Singleton<IDMgr>::Get()->TryUpdateIdAndTaskIndex(&id_state);
  Singleton<TaskStreamIndexManager>::Get()->TryUpdateTaskStreamIndex(id_state.stream_index_state_);
  Singleton<LazyJobBuildAndInferCtxMgr>::Get()->TryUpdateJobIdCount(id_state.job_id_state_);
}

}  // namespace oneflow