diff options
author | Finn Williams <Finn.Williams@arm.com> | 2020-05-15 18:41:05 +0100 |
---|---|---|
committer | finn.williams <finn.williams@arm.com> | 2020-05-19 11:45:57 +0000 |
commit | 45a736263e117b484528e6dee05c549beae721b4 (patch) | |
tree | c674cd5bcd8e22c218c50fadf4c724691c415210 /include | |
parent | b98032f6a46072fee9a2bfcbe631b8193ced567d (diff) | |
download | armnn-45a736263e117b484528e6dee05c549beae721b4.tar.gz |
IVGCVSW-4834 Add calls to increment REGISTERED_BACKENDS and UNREGISTERED_BACKENDS
Signed-off-by: Finn Williams <Finn.Williams@arm.com>
Change-Id: I3600dd15f97ccd4ab745deb87d06ba978e2a0b11
Diffstat (limited to 'include')
-rw-r--r-- | include/armnn/BackendRegistry.hpp | 7 |
1 files changed, 7 insertions, 0 deletions
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(); |