From 344302581b66677a748a456f370752db75adde21 Mon Sep 17 00:00:00 2001 From: Jim Flynn Date: Fri, 4 Mar 2022 15:03:58 +0000 Subject: IVGCVSW-6816 Inject counter registration into ProfilingService Change-Id: I87ce3a1306eced9fc347cc383d9c7bc8994f0b0c Signed-off-by: Jim Flynn --- src/armnn/Runtime.hpp | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) (limited to 'src/armnn/Runtime.hpp') diff --git a/src/armnn/Runtime.hpp b/src/armnn/Runtime.hpp index a8fa5fd5c5..f2462b15d2 100644 --- a/src/armnn/Runtime.hpp +++ b/src/armnn/Runtime.hpp @@ -14,6 +14,7 @@ #include +#include #include #include @@ -24,8 +25,9 @@ namespace armnn { using LoadedNetworks = std::unordered_map>; using IReportStructure = arm::pipe::IReportStructure; + using IInitialiseProfilingService = arm::pipe::IInitialiseProfilingService; -struct RuntimeImpl final : public IReportStructure +struct RuntimeImpl final : public IReportStructure, public IInitialiseProfilingService { public: /// Loads a complete network into the Runtime. @@ -108,7 +110,9 @@ public: //NOTE: we won't need the profiling service reference but it is good to pass the service // in this way to facilitate other implementations down the road - void ReportStructure(); + void ReportStructure() override; + + void InitialiseProfilingService(arm::pipe::IProfilingService& profilingService) override; private: friend void RuntimeLoadedNetworksReserve(RuntimeImpl* runtime); // See RuntimeTests.cpp -- cgit v1.2.1