diff options
| -rw-r--r-- | src/armnn/Runtime.cpp | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/src/armnn/Runtime.cpp b/src/armnn/Runtime.cpp index 63f39c6529..aeecbfedc1 100644 --- a/src/armnn/Runtime.cpp +++ b/src/armnn/Runtime.cpp @@ -322,8 +322,10 @@ Status Runtime::EnqueueWorkload(NetworkId networkId, const InputTensors& inputTensors, const OutputTensors& outputTensors) { - ARMNN_SCOPED_PROFILING_EVENT(Compute::Undefined, "EnqueueWorkload"); LoadedNetwork* loadedNetwork = GetLoadedNetworkPtr(networkId); + ProfilerManager::GetInstance().RegisterProfiler(loadedNetwork->GetProfiler().get()); + + ARMNN_SCOPED_PROFILING_EVENT(Compute::Undefined, "EnqueueWorkload"); static thread_local NetworkId lastId = networkId; if (lastId != networkId) |