Method Init

paddle/fluid/framework/new_executor/executor_statistics.cc:126–167

}

int StatisticsEngine::Init(const platform::NodeTrees& trees) {
  if (inited_) {
    LOG(WARNING) << "Duplicate initialization for StatisticsEngine";
    return -1;
  }
  if (phi::GetCurrentThreadName() != "MainThread") {
    LOG(WARNING) << "StatisticsEngin must run on the main thread";
    return -1;
  }
  inited_ = true;
  InitStdEvents();
  InitInnerthreadPriorityForStdEvents();
  InitInterthreadPriorityForStdEvents();
  // determine executor type
  uint64_t main_tid = phi::GetCurrentThreadId().sys_tid;
  for (const auto& kv : trees.GetNodeTrees()) {
    if (kv.first != main_tid) {
      continue;
    }
    std::queue<const platform::HostTraceEventNode*> q;
    q.push(kv.second);
    while (!q.empty()) {
      auto cur_node = q.front();
      q.pop();
      const auto& name = cur_node->Name();
      if (name.find("Executor::") == 0) {
        VLOG(10) << "type: Executor";
        executor_type_ = ExecutorType::EXECUTOR;
        return InitFiltersForExecutor();
      } else if (name.find("StandaloneExecutor::") == 0) {
        VLOG(10) << "type: InterpreterCore";
        executor_type_ = ExecutorType::INTERPRETER_CORE;
        return InitFiltersForInterpreterCore();
      }
      for (const auto& child : cur_node->GetChildren()) {
        q.push(child);
      }
    }
  }
  LOG(WARNING) << "Unsupported Executor";
  return -1;
}

void StatisticsEngine::InitStdEvents() {
  name2idx_["Total"] = names_.size();