From 01097941ef85073c56cbd1d5f00d7e8ffeb9876d Mon Sep 17 00:00:00 2001 From: Finn Williams Date: Mon, 26 Apr 2021 12:06:34 +0100 Subject: IVGCVSW-5843 Separate memory managers for WorkingMemHandles * Add inter layer memory management to WorkingMemHandle * Change Const layers to be executed once in loadedNetworkConstruction and share tensorHandle between all WorkingMemHandles * Fix various reference workloads pointing to memory in the queueDescriptor Signed-off-by: Finn Williams Change-Id: I69d4b3c5c84d2f5abe4540c3e624ab4f00d88226 --- src/backends/backendsCommon/TensorHandleFactoryRegistry.hpp | 5 +++++ 1 file changed, 5 insertions(+) (limited to 'src/backends/backendsCommon') diff --git a/src/backends/backendsCommon/TensorHandleFactoryRegistry.hpp b/src/backends/backendsCommon/TensorHandleFactoryRegistry.hpp index f926478432..e9e76e73a6 100644 --- a/src/backends/backendsCommon/TensorHandleFactoryRegistry.hpp +++ b/src/backends/backendsCommon/TensorHandleFactoryRegistry.hpp @@ -41,6 +41,11 @@ public: /// Release memory required for inference void ReleaseMemory(); + std::vector>& GetMemoryManagers() + { + return m_MemoryManagers; + } + private: std::vector> m_Factories; std::vector> m_MemoryManagers; -- cgit v1.2.1