Method ThreadGlobalIdGuard

oneflow/core/thread/thread_global_id.cpp:35–43

int64_t GetThisThreadGlobalId() { return *MutThreadLocalUniqueGlobalId(); }

ThreadGlobalIdGuard::ThreadGlobalIdGuard(int64_t thread_global_id)
    : old_thread_global_id_(GetThisThreadGlobalId()) {
  if (old_thread_global_id_ != kThreadGlobalIdMain) {
    CHECK_EQ(old_thread_global_id_, thread_global_id)
        << "nested ThreadGlobalIdGuard disabled. old thread_global_id: " << old_thread_global_id_
        << ", new thread_global_id:" << thread_global_id;
  }
  *MutThreadLocalUniqueGlobalId() = thread_global_id;
}

ThreadGlobalIdGuard::~ThreadGlobalIdGuard() {
  *MutThreadLocalUniqueGlobalId() = old_thread_global_id_;