diff options
Diffstat (limited to 'src/profiling/ConnectionAcknowledgedCommandHandler.hpp')
-rw-r--r-- | src/profiling/ConnectionAcknowledgedCommandHandler.hpp | 11 |
1 files changed, 9 insertions, 2 deletions
diff --git a/src/profiling/ConnectionAcknowledgedCommandHandler.hpp b/src/profiling/ConnectionAcknowledgedCommandHandler.hpp index 6054306da8..e2bdff8e96 100644 --- a/src/profiling/ConnectionAcknowledgedCommandHandler.hpp +++ b/src/profiling/ConnectionAcknowledgedCommandHandler.hpp @@ -5,11 +5,13 @@ #pragma once +#include <armnn/backends/profiling/IBackendProfilingContext.hpp> #include "CommandHandlerFunctor.hpp" #include "ISendCounterPacket.hpp" #include "armnn/profiling/ISendTimelinePacket.hpp" #include "Packet.hpp" #include "ProfilingStateMachine.hpp" +#include <future> namespace armnn { @@ -20,6 +22,9 @@ namespace profiling class ConnectionAcknowledgedCommandHandler final : public CommandHandlerFunctor { +typedef const std::unordered_map<BackendId, std::shared_ptr<armnn::profiling::IBackendProfilingContext>>& + BackendProfilingContexts; + public: ConnectionAcknowledgedCommandHandler(uint32_t familyId, uint32_t packetId, @@ -27,12 +32,14 @@ public: ICounterDirectory& counterDirectory, ISendCounterPacket& sendCounterPacket, ISendTimelinePacket& sendTimelinePacket, - ProfilingStateMachine& profilingStateMachine) + ProfilingStateMachine& profilingStateMachine, + Optional<BackendProfilingContexts> backendProfilingContexts = EmptyOptional()) : CommandHandlerFunctor(familyId, packetId, version) , m_CounterDirectory(counterDirectory) , m_SendCounterPacket(sendCounterPacket) , m_SendTimelinePacket(sendTimelinePacket) , m_StateMachine(profilingStateMachine) + , m_BackendProfilingContext(backendProfilingContexts) {} void operator()(const Packet& packet) override; @@ -42,7 +49,7 @@ private: ISendCounterPacket& m_SendCounterPacket; ISendTimelinePacket& m_SendTimelinePacket; ProfilingStateMachine& m_StateMachine; - + Optional<BackendProfilingContexts> m_BackendProfilingContext; }; } // namespace profiling |