commit e155bbf0e9be6b4d7974297585a59207cd89b00a
author Derek Lamberti <derek.lamberti@arm.com> Wed Oct 13 14:32:12 2021 +0100
committer Derek Lamberti <derek.lamberti@arm.com> Fri Oct 15 15:29:43 2021 +0100
tree 78f5d5f4f2a3a8551854fdd46eac7f599e371bd1
parent 827e4bf519ec1e4af2fb0b32d476b8b1c73b877c

Refactor: Profiler moved to Graph

* This is to enable later work to instrument the Optimizer.

Signed-off-by: Derek Lamberti <derek.lamberti@arm.com>
Change-Id: I2cf1fe022e0d100d6d8705adfbb8cab3ffc96a86

src/armnn/Runtime.cpp

diff --git a/src/armnn/Runtime.cpp b/src/armnn/Runtime.cpp
index a54b712..ca28199 100644
--- a/src/armnn/Runtime.cpp
+++ b/src/armnn/Runtime.cpp
@@ -162,6 +162,10 @@
                                 std::string& errorMessage,
                                 const INetworkProperties& networkProperties)
 {
+    // Register the profiler
+    auto profiler = inNetwork->GetProfiler();
+    ProfilerManager::GetInstance().RegisterProfiler(profiler.get());
+
     IOptimizedNetwork* rawNetwork = inNetwork.release();
 
     networkIdOut = GenerateNetworkId();
@@ -250,6 +254,9 @@
         context.second->AfterUnloadNetwork(networkId);
     }
 
+    // Unregister the profiler
+    ProfilerManager::GetInstance().RegisterProfiler(nullptr);
+
     ARMNN_LOG(debug) << "RuntimeImpl::UnloadNetwork(): Unloaded network with ID: " << networkId;
     return Status::Success;
 }