From ec5463d394453d268706ee6213b1c6a7619d4a5d Mon Sep 17 00:00:00 2001 From: Narumol Prangnawarat Date: Fri, 4 Feb 2022 17:50:20 +0000 Subject: 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 Change-Id: I9f59d57e298d4a494569faec3078d66af799f77b --- include/armnn/backends/IBackendContext.hpp | 3 +++ 1 file changed, 3 insertions(+) (limited to 'include') diff --git a/include/armnn/backends/IBackendContext.hpp b/include/armnn/backends/IBackendContext.hpp index ae85b6354b..6fca42d2ed 100644 --- a/include/armnn/backends/IBackendContext.hpp +++ b/include/armnn/backends/IBackendContext.hpp @@ -25,6 +25,9 @@ public: virtual bool BeforeUnloadNetwork(NetworkId networkId) = 0; virtual bool AfterUnloadNetwork(NetworkId networkId) = 0; + // After Enqueue workload events + virtual bool AfterEnqueueWorkload(NetworkId networkId) = 0; + virtual ~IBackendContext() {} }; -- cgit v1.2.1