commit ec5463d394453d268706ee6213b1c6a7619d4a5d
author Narumol Prangnawarat <narumol.prangnawarat@arm.com> Fri Feb 04 17:50:20 2022 +0000
committer TeresaARM <teresa.charlinreyes@arm.com> Sun Feb 06 19:20:11 2022 +0000
tree 98d17988da24affbe57a324241d9407ae3345f83
parent 23c26277086c78704a17f0dae86da947816320c0

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

src/armnn/Runtime.cpp

diff --git a/src/armnn/Runtime.cpp b/src/armnn/Runtime.cpp
index 95fb8a3..1abe0f3 100644
--- a/src/armnn/Runtime.cpp
+++ b/src/armnn/Runtime.cpp
@@ -242,6 +242,7 @@
                                            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 @@
     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;
 }