diff options
Diffstat (limited to 'tests/profiling/gatordmock/PeriodicCounterCaptureCommandHandler.hpp')
-rw-r--r-- | tests/profiling/gatordmock/PeriodicCounterCaptureCommandHandler.hpp | 19 |
1 files changed, 12 insertions, 7 deletions
diff --git a/tests/profiling/gatordmock/PeriodicCounterCaptureCommandHandler.hpp b/tests/profiling/gatordmock/PeriodicCounterCaptureCommandHandler.hpp index 7d6471bf91..b2fd48f4e8 100644 --- a/tests/profiling/gatordmock/PeriodicCounterCaptureCommandHandler.hpp +++ b/tests/profiling/gatordmock/PeriodicCounterCaptureCommandHandler.hpp @@ -7,6 +7,7 @@ #include <Packet.hpp> #include <CommandHandlerFunctor.hpp> +#include "../../armnn/src/profiling/Packet.hpp" #include <vector> @@ -27,11 +28,15 @@ class PeriodicCounterCaptureCommandHandler : public profiling::CommandHandlerFun { public: - PeriodicCounterCaptureCommandHandler(uint32_t packetId, - uint32_t version, - bool echoPackets) + /** + * + * @param packetId The id of packets this handler will process. + * @param version The version of that id. + * @param quietOperation Optional parameter to turn off printouts. This is useful for unittests. + */ + PeriodicCounterCaptureCommandHandler(uint32_t packetId, uint32_t version, bool quietOperation = false) : CommandHandlerFunctor(packetId, version) - , m_EchoPackets(echoPackets) + , m_QuietOperation(quietOperation) {} void operator()(const armnn::profiling::Packet& packet) override; @@ -46,9 +51,9 @@ private: uint64_t m_FirstTimestamp = 0, m_SecondTimestamp = 0; bool m_HeaderPrinted = false; - bool m_EchoPackets; + bool m_QuietOperation; }; -} // namespace gatordmock +} // namespace gatordmock -} // namespace armnn +} // namespace armnn |