From 45a736263e117b484528e6dee05c549beae721b4 Mon Sep 17 00:00:00 2001
From: Finn Williams <Finn.Williams@arm.com>
Date: Fri, 15 May 2020 18:41:05 +0100
Subject: IVGCVSW-4834 Add calls to increment REGISTERED_BACKENDS and
 UNREGISTERED_BACKENDS

Signed-off-by: Finn Williams <Finn.Williams@arm.com>
Change-Id: I3600dd15f97ccd4ab745deb87d06ba978e2a0b11
---
 include/armnn/BackendRegistry.hpp | 7 +++++++
 1 file changed, 7 insertions(+)

(limited to 'include/armnn')

diff --git a/include/armnn/BackendRegistry.hpp b/include/armnn/BackendRegistry.hpp
index 1aaa11c55c..fe6451cde0 100644
--- a/include/armnn/BackendRegistry.hpp
+++ b/include/armnn/BackendRegistry.hpp
@@ -6,6 +6,7 @@
 
 #include <armnn/Types.hpp>
 #include <armnn/BackendId.hpp>
+#include <armnn/Optional.hpp>
 
 #include <memory>
 #include <unordered_map>
@@ -14,6 +15,10 @@
 namespace armnn
 {
 
+namespace profiling
+{
+    class ProfilingService;
+}
 class IBackendInternal;
 using IBackendInternalUniquePtr = std::unique_ptr<IBackendInternal>;
 
@@ -29,6 +34,7 @@ public:
     size_t Size() const;
     BackendIdSet GetBackendIds() const;
     std::string GetBackendIdsAsString() const;
+    void SetProfilingService(armnn::Optional<profiling::ProfilingService&> profilingService);
 
     BackendRegistry() {}
     virtual ~BackendRegistry() {}
@@ -56,6 +62,7 @@ private:
     BackendRegistry& operator=(const BackendRegistry&) = delete;
 
     FactoryStorage m_Factories;
+    armnn::Optional<profiling::ProfilingService&> m_ProfilingService;
 };
 
 BackendRegistry& BackendRegistryInstance();
-- 
cgit v1.2.1