diff options
Diffstat (limited to 'src/profiling/ProfilingService.hpp')
| -rw-r--r-- | src/profiling/ProfilingService.hpp | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/src/profiling/ProfilingService.hpp b/src/profiling/ProfilingService.hpp index d4fa85604c..1afcb1ca97 100644 --- a/src/profiling/ProfilingService.hpp +++ b/src/profiling/ProfilingService.hpp @@ -16,6 +16,7 @@ #include "ConnectionAcknowledgedCommandHandler.hpp" #include "RequestCounterDirectoryCommandHandler.hpp" #include "PeriodicCounterSelectionCommandHandler.hpp" +#include "PerJobCounterSelectionCommandHandler.hpp" namespace armnn { @@ -96,6 +97,7 @@ private: ConnectionAcknowledgedCommandHandler m_ConnectionAcknowledgedCommandHandler; RequestCounterDirectoryCommandHandler m_RequestCounterDirectoryCommandHandler; PeriodicCounterSelectionCommandHandler m_PeriodicCounterSelectionCommandHandler; + PerJobCounterSelectionCommandHandler m_PerJobCounterSelectionCommandHandler; protected: // Default constructor/destructor kept protected for testing @@ -131,6 +133,9 @@ protected: *this, m_SendCounterPacket, m_StateMachine) + , m_PerJobCounterSelectionCommandHandler(5, + m_PacketVersionResolver.ResolvePacketVersion(4).GetEncodedValue(), + m_StateMachine) { // Register the "Connection Acknowledged" command handler m_CommandHandlerRegistry.RegisterFunctor(&m_ConnectionAcknowledgedCommandHandler); @@ -140,6 +145,9 @@ protected: // Register the "Periodic Counter Selection" command handler m_CommandHandlerRegistry.RegisterFunctor(&m_PeriodicCounterSelectionCommandHandler); + + // Register the "Per-Job Counter Selection" command handler + m_CommandHandlerRegistry.RegisterFunctor(&m_PerJobCounterSelectionCommandHandler); } ~ProfilingService() = default; |