Method CreateOpNode2TaskIds

oneflow/core/graph/task_graph.cpp:1113–1135

}

void BoxingTaskGraph::CreateOpNode2TaskIds(
    const std::function<void(size_t, const std::function<void(size_t i)>&)>& ParallelRunLoop) {
  const OpGraph* op_graph = Singleton<OpGraph>::Get();
  std::vector<const OpNode*> op_nodes;
  op_nodes.reserve(op_graph->node_num());
  op_graph->ForEachNode([&](OpNode* op_node) {
    if (boxing_related_op_node2sorted_comp_tasks_.count(op_node) == 0) {
      op_nodes.push_back(op_node);
      boxing_unrelated_op_node2sorted_task_ids_[op_node].reserve(
          op_node->parallel_desc().parallel_num());
    }
  });
  ParallelRunLoop(op_nodes.size(), [&](size_t i) {
    const OpNode* op_node = op_nodes.at(i);
    TaskType task_type = TaskType4OpNode(op_node);
    const auto& parallel_desc = op_node->parallel_desc();
    auto* task_ids = &boxing_unrelated_op_node2sorted_task_ids_[op_node];
    for (int parallel_id = 0; parallel_id < parallel_desc.parallel_num(); ++parallel_id) {
      const auto& stream_id = GetStreamId(op_node, parallel_id, task_type);
      task_ids->push_back(Singleton<IDMgr>::Get()->GetTaskIdGenerator()->Generate(stream_id));
    }
  });
}

namespace {