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 --- src/armnn/OutputHandler.cpp | 9 +++++++++ 1 file changed, 9 insertions(+) (limited to 'src/armnn/OutputHandler.cpp') diff --git a/src/armnn/OutputHandler.cpp b/src/armnn/OutputHandler.cpp index 807262e482..8827d8ac58 100644 --- a/src/armnn/OutputHandler.cpp +++ b/src/armnn/OutputHandler.cpp @@ -35,4 +35,13 @@ void OutputHandler::CollectWorkloadOutputs(WorkloadDataCollector& dataCollector) dataCollector.Push(m_TensorHandle.get(), m_TensorInfo); } +void OutputHandler::SetAllocatedData() +{ + // Set allocated data only once + if (!m_AllocatedTensorHandle) + { + m_AllocatedTensorHandle = std::move(m_TensorHandle); + } +} + } // namespace armnn -- cgit v1.2.1