diff options
author | Jim Flynn <jim.flynn@arm.com> | 2022-03-04 15:03:58 +0000 |
---|---|---|
committer | Jim Flynn <jim.flynn@arm.com> | 2022-03-06 19:12:00 +0000 |
commit | 344302581b66677a748a456f370752db75adde21 (patch) | |
tree | 6463b11c70c501e46a2a93d7d57f4f68e4b78cb6 /src/armnn/Runtime.hpp | |
parent | 9c315305aaa31a3d1610738cd40f558381f8c687 (diff) | |
download | armnn-344302581b66677a748a456f370752db75adde21.tar.gz |
IVGCVSW-6816 Inject counter registration into ProfilingService
Change-Id: I87ce3a1306eced9fc347cc383d9c7bc8994f0b0c
Signed-off-by: Jim Flynn <jim.flynn@arm.com>
Diffstat (limited to 'src/armnn/Runtime.hpp')
-rw-r--r-- | src/armnn/Runtime.hpp | 8 |
1 files changed, 6 insertions, 2 deletions
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 <armnn/backends/DynamicBackend.hpp> +#include <IInitialiseProfilingService.hpp> #include <IProfilingService.hpp> #include <IReportStructure.hpp> @@ -24,8 +25,9 @@ namespace armnn { using LoadedNetworks = std::unordered_map<NetworkId, std::unique_ptr<LoadedNetwork>>; 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 |