From 55a8ffda24fff5515803df10fb4863d46a1effdf Mon Sep 17 00:00:00 2001
From: Mike Kelly <mike.kelly@arm.com>
Date: Wed, 7 Apr 2021 20:10:49 +0100
Subject: 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
---
 src/armnn/WorkingMemHandle.hpp | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

(limited to 'src/armnn/WorkingMemHandle.hpp')

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;
-- 
cgit v1.2.1