diff options
| author | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2022-02-04 17:50:20 +0000 |
|---|---|---|
| committer | TeresaARM <teresa.charlinreyes@arm.com> | 2022-02-06 19:20:11 +0000 |
| commit | ec5463d394453d268706ee6213b1c6a7619d4a5d (patch) | |
| tree | 98d17988da24affbe57a324241d9407ae3345f83 /src/armnn/Runtime.cpp | |
| parent | 23c26277086c78704a17f0dae86da947816320c0 (diff) | |
| download | armnn-ec5463d394453d268706ee6213b1c6a7619d4a5d.tar.gz | |
IVGCVSW-6747 Call Cl sync after EnqueueWorkload
* Add AfterEnqueueWorkload to IBackendContext
* Implement AfterEnqueueWorkload in ClBackendContext to call Cl sync
* Set allocated data on outputhandler only once
* Handle PreImportedHandles and CurImportedId the same way as Async
Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: I9f59d57e298d4a494569faec3078d66af799f77b
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; } |