Method Prerun

executor/engine/generic_engine.cpp:76–125

}

bool GenericEngine::Prerun(exec_handle_t h)
{
    GraphTask* graph_task = any_cast<GraphTask*>(*h);
    GraphExecutor* graph_executor = graph_task->GetGraphExecutor();
    Graph* graph = graph_task->GetGraph();

    // call DevAllocator to allocate graph into

    DevAllocator* dev_allocator = nullptr;

    std::string alloc_policy;

    TEngineConfig::Get("dev_allocator", alloc_policy);

    if(alloc_policy.empty() || !DevAllocatorManager::Get(alloc_policy, dev_allocator))
    {
        XLOG_ERROR() << "cannot get proper dev allocator\n";
        return false;
    }

    std::vector<Subgraph*> sub_graph_list;

    if(!dev_allocator->Allocate(this, graph_executor, graph, sub_graph_list))
    {
        XLOG_ERROR() << "dev executor allocator failed\n";

        for(auto e : sub_graph_list)
            delete e;

        return false;
    }

    // create SubgraphTask
    for(unsigned int i = 0; i < sub_graph_list.size(); i++)
    {
        Subgraph* sub_graph = sub_graph_list[i];
        SubgraphTask* new_task = new SubgraphTask(sub_graph);

        new_task->Init(graph_task);
        graph_task->AddSubgraphTask(new_task);
    }

    if(!graph_task->Prerun())
    {
        graph_task->ReclaimSubgraphTask();
        return false;
    }

    return true;
}

bool GenericEngine::Run(exec_handle_t h, exec_event_t& event)
{