Method AllocateMem

driver/cpu/cpu_runner.cpp:851–980

}

bool CPURunner::AllocateMem(Subgraph* sub_graph)
{
    const std::vector<Node*>& seq_nodes = sub_graph->seq_nodes;

    /*
       first, check if any nodes supports new memory interface
       this memory block is only for tempory use and so that it can be shared
       between operators
     */

    unsigned int max_shared_mem_size = 0;

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

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

        NodeOps* node_ops = any_cast<NodeOps*>(node->GetAttr(ATTR_NODE_OPS));
        unsigned int mem_size = 0;

        if(node_ops->GetSharedMemorySize(node, mem_size) && mem_size > max_shared_mem_size)
        {
            max_shared_mem_size = mem_size;
        }
    }

    if(max_shared_mem_size > 0)
    {
        void* shared_memory = mem_alloc(max_shared_mem_size + 128+MEM_ALIGN_SIZE);
        sub_graph->SetAttr("shared_temp_memory", shared_memory);

        shared_memory=(void *)(((long)(shared_memory)+MEM_ALIGN_SIZE-1)&(MEM_ALIGN_MASK));

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

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

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

            unsigned int mem_size = 0;

            if(node_ops->GetSharedMemorySize(node, mem_size))
                node_ops->SetSharedMemoryAddr(node, shared_memory, mem_size);
        }

        // std::cout<<"max shared memory: "<<max_shared_mem_size<<"\n";
    }

    /*
     *  now, calculate the maximum input and output memory blocks to run the graph
     */

    std::vector<int> mem_blocks;