From e155bbf0e9be6b4d7974297585a59207cd89b00a Mon Sep 17 00:00:00 2001 From: Derek Lamberti Date: Wed, 13 Oct 2021 14:32:12 +0100 Subject: Refactor: Profiler moved to Graph * This is to enable later work to instrument the Optimizer. Signed-off-by: Derek Lamberti Change-Id: I2cf1fe022e0d100d6d8705adfbb8cab3ffc96a86 --- src/armnn/LoadedNetwork.cpp | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) (limited to 'src/armnn/LoadedNetwork.cpp') diff --git a/src/armnn/LoadedNetwork.cpp b/src/armnn/LoadedNetwork.cpp index c161ed35d5..4688b6eea4 100644 --- a/src/armnn/LoadedNetwork.cpp +++ b/src/armnn/LoadedNetwork.cpp @@ -122,13 +122,13 @@ LoadedNetwork::LoadedNetwork(std::unique_ptr net, m_TensorHandleFactoryRegistry(), m_ProfilingService(profilingService) { - // Create a profiler and register it for the current thread. - m_Profiler = std::make_shared(); - ProfilerManager::GetInstance().RegisterProfiler(m_Profiler.get()); + // Get the profiler and register it for the current thread. + const std::shared_ptr& profiler = m_OptimizedNetwork->GetProfiler(); + ProfilerManager::GetInstance().RegisterProfiler(profiler.get()); - m_Profiler->EnableProfiling(networkProperties.m_ProfilingEnabled); + profiler->EnableProfiling(networkProperties.m_ProfilingEnabled); - m_Profiler->EnableNetworkDetailsToStdOut(networkProperties.m_OutputNetworkDetailsMethod); + profiler->EnableNetworkDetailsToStdOut(networkProperties.m_OutputNetworkDetailsMethod); Graph& order = m_OptimizedNetwork->pOptimizedNetworkImpl->GetGraph().TopologicalSort(); //First create tensor handlers, backends and workload factories. -- cgit v1.2.1