From 72219330fd85b1271e714d4ba894d6d8e26340c9 Mon Sep 17 00:00:00 2001 From: Georgios Pinitas Date: Tue, 5 Jun 2018 14:56:06 +0100 Subject: COMPMID-1145: (API) Introduce prepare() stage (NEON/CL/GLES) Change-Id: I5b46764f9c3154ec3e3b9c951cc9e6dfbcb81dfb Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/134255 Reviewed-by: Anthony Barbier Tested-by: Jenkins Reviewed-by: Pablo Tello Reviewed-by: Michele DiGiorgio --- src/graph/GraphManager.cpp | 18 ++---------------- 1 file changed, 2 insertions(+), 16 deletions(-) (limited to 'src/graph') diff --git a/src/graph/GraphManager.cpp b/src/graph/GraphManager.cpp index 0ea3254faa..10661ea275 100644 --- a/src/graph/GraphManager.cpp +++ b/src/graph/GraphManager.cpp @@ -76,12 +76,8 @@ void GraphManager::finalize_graph(Graph &graph, GraphContext &ctx, PassManager & detail::allocate_const_tensors(graph); detail::call_all_const_node_accessors(graph); - // TODO (COMPMID-920) : Update prepare for NEON/GC - if(forced_target == Target::CL) - { - // Prepare graph - detail::prepare_all_tasks(workload); - } + // Prepare graph + detail::prepare_all_tasks(workload); // Setup tensor memory (Allocate all tensors or setup transition manager) if(ctx.config().use_transition_memory_manager) @@ -99,16 +95,6 @@ void GraphManager::finalize_graph(Graph &graph, GraphContext &ctx, PassManager & // Register graph _workloads.insert(std::make_pair(graph.id(), std::move(workload))); ARM_COMPUTE_LOG_GRAPH_VERBOSE("Created workload for graph with ID : " << graph.id().get() << std::endl); - - // TODO (COMPMID-920) : Update prepare for NEON/GC - if(forced_target != Target::CL) - { - // Make first run - execute_graph(graph); - - // Release all unused const tensors - detail::release_unused_tensors(graph); - } } void GraphManager::execute_graph(Graph &graph) -- cgit v1.2.1