From c4946c7addd65f032951dbf16fb824cdd67fd463 Mon Sep 17 00:00:00 2001 From: janeil01 Date: Thu, 7 Nov 2019 09:32:28 +0000 Subject: IVGCVSW-4102 Move ProfilingGuid to public interface * Moved ProfilingGuid to Types.hpp * Refactoring to enable ProfilingGuid Signed-off-by: janeil01 Change-Id: Ibf77002d74e484f8a63ffd96aa14303c1f0d38ae --- include/armnn/Types.hpp | 93 +++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 93 insertions(+) (limited to 'include') diff --git a/include/armnn/Types.hpp b/include/armnn/Types.hpp index 51162e6cf3..b2aa52edb2 100644 --- a/include/armnn/Types.hpp +++ b/include/armnn/Types.hpp @@ -7,6 +7,7 @@ #include #include #include +#include #include "BackendId.hpp" #include "Exceptions.hpp" @@ -219,4 +220,96 @@ class ITensorHandle; /// @param tensorHandle - TensorHandle for the input tensor to the Debug layer using DebugCallbackFunction = std::function; + +namespace profiling +{ + +class ProfilingGuid +{ +public: + ProfilingGuid(uint64_t guid) : m_Guid(guid) {} + + operator uint64_t() const { return m_Guid; } + + bool operator==(const ProfilingGuid& other) const + { + return m_Guid == other.m_Guid; + } + + bool operator!=(const ProfilingGuid& other) const + { + return m_Guid != other.m_Guid; + } + + bool operator<(const ProfilingGuid& other) const + { + return m_Guid < other.m_Guid; + } + + bool operator<=(const ProfilingGuid& other) const + { + return m_Guid <= other.m_Guid; + } + + bool operator>(const ProfilingGuid& other) const + { + return m_Guid > other.m_Guid; + } + + bool operator>=(const ProfilingGuid& other) const + { + return m_Guid >= other.m_Guid; + } + +protected: + uint64_t m_Guid; +}; + +/// Strongly typed guids to distinguish between those generated at runtime, and those that are statically defined. +struct ProfilingDynamicGuid : public ProfilingGuid +{ + using ProfilingGuid::ProfilingGuid; +}; + +struct ProfilingStaticGuid : public ProfilingGuid +{ + using ProfilingGuid::ProfilingGuid; +}; + +} // namespace profiling + } // namespace armnn + + +namespace std +{ +// make ProfilingGuid hashable +template<> +struct hash +{ + std::size_t operator()(armnn::profiling::ProfilingGuid const& guid) const noexcept + { + return hash()(uint64_t(guid)); + } +}; + +// make ProfilingDynamicGuid hashable +template<> +struct hash +{ + std::size_t operator()(armnn::profiling::ProfilingDynamicGuid const& guid) const noexcept + { + return hash()(uint64_t(guid)); + } +}; + +// make ProfilingStaticGuid hashable +template<> +struct hash +{ + std::size_t operator()(armnn::profiling::ProfilingStaticGuid const& guid) const noexcept + { + return hash()(uint64_t(guid)); + } +}; +} -- cgit v1.2.1