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