aboutsummaryrefslogtreecommitdiff
path: root/profiling
diff options
context:
space:
mode:
authorNikhil Raj <nikhil.raj@arm.com>2021-06-08 12:31:50 +0100
committerNikhil Raj <nikhil.raj@arm.com>2021-06-08 16:04:14 +0100
commit5b1bcc93820b442bc4035c1e030a8d4a0983df91 (patch)
tree42b8cc8bcbd656ed9ac01fa18232867fce9959b3 /profiling
parent400c593e03af3cc9131058b3cb30b787586f69a3 (diff)
downloadarmnn-5b1bcc93820b442bc4035c1e030a8d4a0983df91.tar.gz
IVGCVSW-5834 Move the IProfilingGuidGenerator and ProfilingGuidGenerator into profiling common
Signed-off-by: Nikhil Raj <nikhil.raj@arm.com> Change-Id: I0d672cc782cc2de66a88acf0d83fcd40208ace95
Diffstat (limited to 'profiling')
-rw-r--r--profiling/common/include/IProfilingGuidGenerator.hpp32
-rw-r--r--profiling/common/include/ProfilingGuidGenerator.hpp62
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