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.hpp | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) (limited to 'src/armnn/OutputHandler.hpp') diff --git a/src/armnn/OutputHandler.hpp b/src/armnn/OutputHandler.hpp index 3fd2519ed5..d1cb2deea3 100644 --- a/src/armnn/OutputHandler.hpp +++ b/src/armnn/OutputHandler.hpp @@ -50,15 +50,15 @@ public: void SetData(std::unique_ptr data) { m_TensorHandle = std::move(data); } - void SetAllocatedData() { m_AllocatedTensorHandle = std::move(m_TensorHandle); } + void SetAllocatedData(); - void UseAllocatedData() { m_TensorHandle = std::move(m_AllocatedTensorHandle); } + void UseAllocatedData() { m_TensorHandle = m_AllocatedTensorHandle; } /// @brief Returns true if SetTensorInfo() has been called at least once on this. bool IsTensorInfoSet() const { return m_bTensorInfoSet; } private: - std::unique_ptr m_TensorHandle; - std::unique_ptr m_AllocatedTensorHandle; + std::shared_ptr m_TensorHandle; + std::shared_ptr m_AllocatedTensorHandle; TensorInfo m_TensorInfo; bool m_bTensorInfoSet = false; }; -- cgit v1.2.1