Method Init

executor/lib/graph_task.cpp:629–701

}

void SubgraphTask::Init(GraphTask* graph_tsk)
{
    graph_task = graph_tsk;
    Graph* graph = graph_task->GetGraph();
    const ExecAttr* p_exec_attr = graph_task->GetExecAttr();

    exec_priority = p_exec_attr->priority;

    // check if it is a global output task
    for(auto e : sub_graph->output_nodes)
    {
        for(auto m : graph->output_nodes)
        {
            if(e == m)
            {
                is_output_task = true;
                break;
            }
        }
    }

    saved_input_wait_count_ = 0;

    // for input nodes, calculate the input_wait_mask
    for(Node* node : sub_graph->input_nodes)
    {
        std::uint64_t mask = 0;

        for(unsigned int i = 0; i < node->GetInputNum(); i++)
        {
            NodePort* port = node->GetInputPort(i);
            Tensor* tensor = port->tensor;

            if(tensor->GetType() == kVarTensor)
            {
                /* check if parent is in the same graph or not */
                Node* parent = tensor->producer->owner;
                bool parent_in_graph = false;

                for(unsigned int i = 0; i < sub_graph->seq_nodes.size(); i++)
                {
                    if(parent == sub_graph->seq_nodes[i])
                    {
                        parent_in_graph = true;
                        break;
                    }
                }

                if(!parent_in_graph)
                    mask |= 1 << port->port_index;
            }
        }

        if(mask)
        {
            SetNodeInputWaitMask(node, mask);
            CreateNodeInputWaitCounter(node);
            auto p_counter = GetNodeInputWaitCounter(node);