24 #include "arm_compute/graph/GraphManager.h"
26 #include "arm_compute/graph/Graph.h"
27 #include "arm_compute/graph/GraphContext.h"
28 #include "arm_compute/graph/Logger.h"
29 #include "arm_compute/graph/PassManager.h"
30 #include "arm_compute/graph/TypePrinter.h"
31 #include "arm_compute/graph/Utils.h"
32 #include "arm_compute/graph/algorithms/TopologicalSort.h"
33 #include "arm_compute/graph/detail/CrossLayerMemoryManagerHelpers.h"
34 #include "arm_compute/graph/detail/ExecutionHelpers.h"
40 namespace graph  namespace
47 void GraphManager::finalize_graph(Graph &graph, GraphContext &ctx, PassManager &pm, Target target)  in finalize_graph()  argument
49     ARM_COMPUTE_LOG_INFO_WITH_FUNCNAME_ACL("Initiate graph configuration!");  in finalize_graph()
51     // Check if graph has been registered  in finalize_graph()
52     if(_workloads.find(graph.id()) != std::end(_workloads))  in finalize_graph()
54         ARM_COMPUTE_ERROR("Graph is already registered!");  in finalize_graph()
58     pm.run_type(graph, IGraphMutator::MutationType::IR);  in finalize_graph()
60     // Force target to all graph construct  in finalize_graph()
79     force_target_to_graph(graph, forced_target);  in finalize_graph()
85     detail::configure_all_tensors(graph);  in finalize_graph()
88     pm.run_type(graph, IGraphMutator::MutationType::Backend);  in finalize_graph()
91     std::vector<NodeID> topological_sorted_nodes = dfs(graph);  in finalize_graph()
94     detail::validate_all_nodes(graph);  in finalize_graph()
97     auto workload = detail::configure_all_nodes(graph, ctx, topological_sorted_nodes);  in finalize_graph()
101     detail::allocate_const_tensors(graph);  in finalize_graph()
102     detail::call_all_const_node_accessors(graph);  in finalize_graph()
104     // Prepare graph  in finalize_graph()
110         detail::configure_transition_manager(graph, ctx, workload);  in finalize_graph()
114         detail::allocate_all_tensors(graph);  in finalize_graph()
117     // Finalize Graph context  in finalize_graph()
120     // Register graph  in finalize_graph()
121     _workloads.insert(std::make_pair(graph.id(), std::move(workload)));  in finalize_graph()
122 … ARM_COMPUTE_LOG_GRAPH_VERBOSE("Created workload for graph with ID : " << graph.id() << std::endl);  in finalize_graph()
125 void GraphManager::execute_graph(Graph &graph)  in execute_graph()  argument
127     ARM_COMPUTE_LOG_INFO_WITH_FUNCNAME_ACL("Initiate graph execution!");  in execute_graph()
129     // Check if graph is finalized  in execute_graph()
130     auto it = _workloads.find(graph.id());  in execute_graph()
131     ARM_COMPUTE_ERROR_ON_MSG(it == std::end(_workloads), "Graph is not registered!");  in execute_graph()
141         // Run graph  in execute_graph()
152 void GraphManager::invalidate_graph(Graph &graph)  in invalidate_graph()  argument
154     auto it = _workloads.find(graph.id());  in invalidate_graph()
155     ARM_COMPUTE_ERROR_ON_MSG(it == std::end(_workloads), "Graph is not registered!");  in invalidate_graph()
159 } // namespace graph