Method SelectTaskNodeByRank

oneflow/core/graph/task_graph.cpp:1175–1182

static*/

}  // namespace

/*static*/ bool BoxingTaskGraph::SelectTaskNodeByRank(TaskNode* task_node, int64_t rank) {
  return TaskNodeVisitor<bool>(
      task_node, [&](TransportTaskNode* task_node) { return task_node->machine_id() == rank; },
      [&](CompTaskNode* task_node) {
        const auto& machine_id = task_node->machine_id();
        return IsComputTaskNodeDutyRank(rank, task_node->op_node()->parallel_desc(), machine_id);
      });
}

void BoxingTaskGraph::ToProto(const std::function<bool(TaskNode*)>& Pick,
                              BoxingTaskGraphProto* proto) const {