Function CalculateMemBlocks

driver/cpu/cpu_runner.cpp:758–849

}

static void CalculateMemBlocks(std::vector<int>& mem_block, Subgraph* sub_graph)
{
    // first calculate max var tensor exists
    tensor_map_t tensor_map;

    const std::vector<Node*>& seq_nodes = sub_graph->seq_nodes;

    int node_number = seq_nodes.size();
    int max_active_num = 0;
    int active_num = 0;

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

        // first, add output tensor into map
        if(!node->IsDynamicShape() && node->ExistAttr(ATTR_NODE_OPS))
        {
            for(unsigned int j = 0; j < node->GetOutputNum(); j++)
            {
                Tensor* tensor = node->GetOutputTensor(j);

                if(get_tensor_mem(tensor))
                    continue;

                int consumer_number = tensor->consumer.size();

                tensor_map[tensor] = consumer_number;
                active_num++;
            }
        }

        if(active_num > max_active_num)
            max_active_num = active_num;

        // second, reduce the active_num by  release input
        for(unsigned int j = 0; j < node->GetInputNum(); j++)
        {
            Tensor* tensor = node->GetInputTensor(j);

            if(tensor_map.count(tensor) == 0)
                continue;

            auto ir = tensor_map.find(tensor);

            ir->second--;

            if(ir->second == 0)
            {
                active_num--;
                tensor_map.erase(ir);
            }
        }
    }

    // suppose each output node only has one output tensor
    if(active_num > ( int )sub_graph->output_nodes.size())
    {