diff options
author | Sadik Armagan <sadik.armagan@arm.com> | 2020-02-17 11:33:31 +0000 |
---|---|---|
committer | Sadik Armagan <sadik.armagan@arm.com> | 2020-02-17 11:33:31 +0000 |
commit | cab588ac45bf1c39832d8a278ee46db851c1b015 (patch) | |
tree | 30b038b1859fb51d7c4ca7e312bfcc7b59ea8ddc /src/profiling/test/ProfilingMocks.hpp | |
parent | c5a680111396235f7b772c99a2028cf76c4e63c8 (diff) | |
download | armnn-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.hpp | 69 |
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 |