1 files changed, 6 insertions, 0 deletions
diff --git a/src/armnn/WorkingMemHandle.hpp b/src/armnn/WorkingMemHandle.hpp
index 92b0acaec3..5ccb2b2342 100644
--- a/src/armnn/WorkingMemHandle.hpp
+++ b/src/armnn/WorkingMemHandle.hpp
@@ -38,6 +38,11 @@ public:
         return m_NetworkId;
     }
 
+    profiling::ProfilingGuid GetInferenceId() override
+    {
+        return m_InferenceId;
+    }
+
     /// 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.
     void Allocate() override;
@@ -87,6 +92,7 @@ private:
 
     bool m_IsAllocated;
     std::mutex m_Mutex;
+    profiling::ProfilingGuid m_InferenceId;
 };
 
 } // end experimental namespace