diff options
Diffstat (limited to 'src/armnn/Runtime.cpp')
| -rw-r--r-- | src/armnn/Runtime.cpp | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/src/armnn/Runtime.cpp b/src/armnn/Runtime.cpp index 95fb8a3abb..1abe0f394b 100644 --- a/src/armnn/Runtime.cpp +++ b/src/armnn/Runtime.cpp @@ -242,6 +242,7 @@ Status RuntimeImpl::UnloadNetwork(NetworkId networkId) profiling::LabelsAndEventClasses::ARMNN_PROFILING_EOL_EVENT_CLASS); } } + if (m_LoadedNetworks.erase(networkId) == 0) { ARMNN_LOG(warning) << "WARNING: RuntimeImpl::UnloadNetwork(): " << networkId << " not found!"; @@ -632,6 +633,12 @@ Status RuntimeImpl::EnqueueWorkload(NetworkId networkId, ARMNN_LOG(info) << "Execution time: " << std::setprecision(2) << std::fixed << armnn::GetTimeDuration(startTime).count() << " ms."; + // Call After EnqueueWorkload events + for (auto&& context : m_BackendContexts) + { + context.second->AfterEnqueueWorkload(networkId); + } + return status; } |