diff options
Diffstat (limited to 'profiling/common')
-rw-r--r-- | profiling/common/include/IProfilingGuidGenerator.hpp | 32 | ||||
-rw-r--r-- | profiling/common/include/ProfilingGuidGenerator.hpp | 62 |
2 files changed, 94 insertions, 0 deletions
diff --git a/profiling/common/include/IProfilingGuidGenerator.hpp b/profiling/common/include/IProfilingGuidGenerator.hpp new file mode 100644 index 0000000000..34ee9673b5 --- /dev/null +++ b/profiling/common/include/IProfilingGuidGenerator.hpp @@ -0,0 +1,32 @@ +// +// Copyright © 2019 Arm Ltd. All rights reserved. +// SPDX-License-Identifier: MIT +// + +#pragma once + +#include "ProfilingGuid.hpp" + +#include <string> + +namespace armnn +{ + +namespace profiling +{ + +class IProfilingGuidGenerator +{ +public: + /// Return the next random Guid in the sequence + virtual ProfilingDynamicGuid NextGuid() = 0; + + /// Create a ProfilingStaticGuid based on a hash of the string + virtual ProfilingStaticGuid GenerateStaticId(const std::string& str) = 0; + + virtual ~IProfilingGuidGenerator() {} +}; + +} // namespace profiling + +} // namespace armnn diff --git a/profiling/common/include/ProfilingGuidGenerator.hpp b/profiling/common/include/ProfilingGuidGenerator.hpp new file mode 100644 index 0000000000..2b7302bd61 --- /dev/null +++ b/profiling/common/include/ProfilingGuidGenerator.hpp @@ -0,0 +1,62 @@ +// +// Copyright © 2019 Arm Ltd and Contributors. All rights reserved. +// SPDX-License-Identifier: MIT +// + +#pragma once + +#include "IProfilingGuidGenerator.hpp" +#include "ProfilingGuid.hpp" + +#include <functional> +#include <mutex> + +namespace armnn +{ + +namespace profiling +{ + +class ProfilingGuidGenerator : public IProfilingGuidGenerator +{ +public: + /// Construct a generator with the default address space static/dynamic partitioning + ProfilingGuidGenerator() : m_Sequence(0) {} + + /// Return the next random Guid in the sequence + inline ProfilingDynamicGuid NextGuid() override + { + std::lock_guard<std::mutex> sequencelock(m_SequenceMutex); + ProfilingDynamicGuid guid(m_Sequence); + m_Sequence++; + if (m_Sequence >= MIN_STATIC_GUID) + { + // Reset the sequence to 0 when it reaches the upper bound of dynamic guid + m_Sequence = 0; + } + return guid; + } + + /// Create a ProfilingStaticGuid based on a hash of the string + inline ProfilingStaticGuid GenerateStaticId(const std::string& str) override + { + uint64_t staticHash = m_Hash(str) | MIN_STATIC_GUID; + return ProfilingStaticGuid(staticHash); + } + + /// Reset the generator back to zero. Used mainly for test. + inline void Reset() + { + std::lock_guard<std::mutex> sequencelock(m_SequenceMutex); + m_Sequence = 0; + } + +private: + std::hash<std::string> m_Hash; + uint64_t m_Sequence; + std::mutex m_SequenceMutex; +}; + +} // namespace profiling + +} // namespace armnn |