aboutsummaryrefslogtreecommitdiff
path: root/src/profiling/test/ProfilingMocks.hpp
diff options
context:
space:
mode:
authorSadik Armagan <sadik.armagan@arm.com>2020-02-17 11:33:31 +0000
committerSadik Armagan <sadik.armagan@arm.com>2020-02-17 11:33:31 +0000
commitcab588ac45bf1c39832d8a278ee46db851c1b015 (patch)
tree30b038b1859fb51d7c4ca7e312bfcc7b59ea8ddc /src/profiling/test/ProfilingMocks.hpp
parentc5a680111396235f7b772c99a2028cf76c4e63c8 (diff)
downloadarmnn-cab588ac45bf1c39832d8a278ee46db851c1b015.tar.gz
IVGCVSW-4320 Implement the bulk reporting of counter values from backends
* Implemented unit test for ReportCounters function in BackendProfiling Signed-off-by: Jim Flynn <jim.flynn@arm.com> Signed-off-by: Sadik Armagan <sadik.armagan@arm.com> Change-Id: Iec342a96060c8ef6090e6cc67bda8a7a3e890c50
Diffstat (limited to 'src/profiling/test/ProfilingMocks.hpp')
-rw-r--r--src/profiling/test/ProfilingMocks.hpp69
1 files changed, 69 insertions, 0 deletions
diff --git a/src/profiling/test/ProfilingMocks.hpp b/src/profiling/test/ProfilingMocks.hpp
index 9d1321345a..3782a0f7e4 100644
--- a/src/profiling/test/ProfilingMocks.hpp
+++ b/src/profiling/test/ProfilingMocks.hpp
@@ -5,6 +5,7 @@
#pragma once
+#include <Holder.hpp>
#include <IProfilingConnectionFactory.hpp>
#include <IProfilingService.hpp>
#include <ProfilingGuidGenerator.hpp>
@@ -647,6 +648,74 @@ private:
Counters m_Counters;
};
+class MockProfilingService : public IProfilingService, public IRegisterCounterMapping
+{
+public:
+ MockProfilingService(MockBufferManager& mockBufferManager,
+ bool isProfilingEnabled,
+ const CaptureData& captureData) :
+ m_SendCounterPacket(mockBufferManager),
+ m_IsProfilingEnabled(isProfilingEnabled),
+ m_CaptureData(captureData) {}
+
+ /// Return the next random Guid in the sequence
+ ProfilingDynamicGuid NextGuid() override
+ {
+ return m_GuidGenerator.NextGuid();
+ }
+
+ /// Create a ProfilingStaticGuid based on a hash of the string
+ ProfilingStaticGuid GenerateStaticId(const std::string& str) override
+ {
+ return m_GuidGenerator.GenerateStaticId(str);
+ }
+
+ std::unique_ptr<ISendTimelinePacket> GetSendTimelinePacket() const override
+ {
+ return nullptr;
+ }
+
+ const ICounterMappings& GetCounterMappings() const override
+ {
+ return m_CounterMapping;
+ }
+
+ ISendCounterPacket& GetSendCounterPacket() override
+ {
+ return m_SendCounterPacket;
+ }
+
+ bool IsProfilingEnabled() const override
+ {
+ return m_IsProfilingEnabled;
+ }
+
+ CaptureData GetCaptureData() override
+ {
+ CaptureData copy(m_CaptureData);
+ return copy;
+ }
+
+ void RegisterMapping(uint16_t globalCounterId,
+ uint16_t backendCounterId,
+ const armnn::BackendId& backendId) override
+ {
+ m_CounterMapping.RegisterMapping(globalCounterId, backendCounterId, backendId);
+ }
+
+ void Reset() override
+ {
+ m_CounterMapping.Reset();
+ }
+
+private:
+ ProfilingGuidGenerator m_GuidGenerator;
+ CounterIdMap m_CounterMapping;
+ SendCounterPacket m_SendCounterPacket;
+ bool m_IsProfilingEnabled;
+ CaptureData m_CaptureData;
+};
+
} // namespace profiling
} // namespace armnn