Method Prerun

driver/cpu/cpu_runner.cpp:237–262

bool debug_graph = false;

bool CPURunner::Prerun(Subgraph* sub_graph)
{
    if(!BindNodeOps(sub_graph))
        return false;

    if(!AllocateMem(sub_graph))
        return false;

    for(unsigned int i = 0; i < sub_graph->seq_nodes.size(); i++)
    {
        Node* node = sub_graph->seq_nodes[i];

        if(!node->ExistAttr(ATTR_NODE_OPS))
            continue;

        NodeOps* node_ops = any_cast<NodeOps*>(node->GetAttr(ATTR_NODE_OPS));

        if(!node_ops->Prerun(node))
        {
            XLOG_ERROR() << "Prerun for node: " << node->GetName() << " op: " << node->GetOp()->GetName() << " failed\n";
            return false;
        }
    }

    return true;
}

#ifdef ENABLE_TIME_PROFILING