diff options
author | Mike Kelly <mike.kelly@arm.com> | 2021-04-07 20:10:49 +0100 |
---|---|---|
committer | finn.williams <finn.williams@arm.com> | 2021-04-08 11:23:47 +0000 |
commit | 55a8ffda24fff5515803df10fb4863d46a1effdf (patch) | |
tree | e314dea48f22ae88d452527b2decaca61df108ad /src/armnn/WorkingMemHandle.hpp | |
parent | b76eaed55a89330b3b448c4f4522b3fc94a4f38d (diff) | |
download | armnn-55a8ffda24fff5515803df10fb4863d46a1effdf.tar.gz |
IVGCVSW-5823 Refactor Async Network API
* Moved IAsyncNetwork into IRuntime.
* All LoadedNetworks can be executed Asynchronously.
Signed-off-by: Mike Kelly <mike.kelly@arm.com>
Change-Id: Ibbc901ab9110dc2f881425b75489bccf9ad54169
Diffstat (limited to 'src/armnn/WorkingMemHandle.hpp')
-rw-r--r-- | src/armnn/WorkingMemHandle.hpp | 12 |
1 files changed, 10 insertions, 2 deletions
diff --git a/src/armnn/WorkingMemHandle.hpp b/src/armnn/WorkingMemHandle.hpp index 090f180206..cef6fb6fd3 100644 --- a/src/armnn/WorkingMemHandle.hpp +++ b/src/armnn/WorkingMemHandle.hpp @@ -24,10 +24,17 @@ class WorkingMemHandle final : public IWorkingMemHandle { public: - WorkingMemHandle(std::vector<WorkingMemDescriptor> workingMemDescriptors, + WorkingMemHandle(NetworkId networkId, + std::vector<WorkingMemDescriptor> workingMemDescriptors, std::unordered_map<LayerGuid, WorkingMemDescriptor> workingMemDescriptorMap); - ~WorkingMemHandle() { FreeWorkingMemory(); } + ~WorkingMemHandle() + { FreeWorkingMemory(); } + + NetworkId GetNetworkId() override + { + return m_NetworkId; + } /// Allocate the backing memory required for execution. If this is not called, then allocation will be /// deferred to execution time. The mutex must be locked. @@ -106,6 +113,7 @@ public: private: void FreeWorkingMemory(); + NetworkId m_NetworkId; std::shared_ptr<ProfilerImpl> m_Profiler; std::vector<WorkingMemDescriptor> m_WorkingMemDescriptors; |