From 5b1bcc93820b442bc4035c1e030a8d4a0983df91 Mon Sep 17 00:00:00 2001
From: Nikhil Raj <nikhil.raj@arm.com>
Date: Tue, 8 Jun 2021 12:31:50 +0100
Subject: IVGCVSW-5834 Move the IProfilingGuidGenerator and
 ProfilingGuidGenerator into profiling common

Signed-off-by: Nikhil Raj <nikhil.raj@arm.com>
Change-Id: I0d672cc782cc2de66a88acf0d83fcd40208ace95
---
 .../common/include/IProfilingGuidGenerator.hpp     | 32 +++++++++++
 .../common/include/ProfilingGuidGenerator.hpp      | 62 ++++++++++++++++++++++
 2 files changed, 94 insertions(+)
 create mode 100644 profiling/common/include/IProfilingGuidGenerator.hpp
 create mode 100644 profiling/common/include/ProfilingGuidGenerator.hpp

(limited to 'profiling')

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
-- 
cgit v1.2.1