| // |
| // Copyright © 2019 Arm Ltd and Contributors. All rights reserved. |
| // SPDX-License-Identifier: MIT |
| // |
| |
| #pragma once |
| |
| #include <armnn/backends/profiling/IBackendProfilingContext.hpp> |
| #include "IProfilingServiceStatus.hpp" |
| #include "ISendCounterPacket.hpp" |
| #include "armnn/profiling/ISendTimelinePacket.hpp" |
| #include "ProfilingStateMachine.hpp" |
| |
| #include <common/include/CommandHandlerFunctor.hpp> |
| #include <common/include/Packet.hpp> |
| |
| namespace armnn |
| { |
| |
| namespace profiling |
| { |
| |
| class ConnectionAcknowledgedCommandHandler final : public arm::pipe::CommandHandlerFunctor |
| { |
| |
| typedef const std::unordered_map<BackendId, std::shared_ptr<armnn::profiling::IBackendProfilingContext>>& |
| BackendProfilingContexts; |
| |
| public: |
| ConnectionAcknowledgedCommandHandler(uint32_t familyId, |
| uint32_t packetId, |
| uint32_t version, |
| ICounterDirectory& counterDirectory, |
| ISendCounterPacket& sendCounterPacket, |
| ISendTimelinePacket& sendTimelinePacket, |
| ProfilingStateMachine& profilingStateMachine, |
| IProfilingServiceStatus& profilingServiceStatus, |
| Optional<BackendProfilingContexts> backendProfilingContexts = EmptyOptional()) |
| : CommandHandlerFunctor(familyId, packetId, version) |
| , m_CounterDirectory(counterDirectory) |
| , m_SendCounterPacket(sendCounterPacket) |
| , m_SendTimelinePacket(sendTimelinePacket) |
| , m_StateMachine(profilingStateMachine) |
| , m_ProfilingServiceStatus(profilingServiceStatus) |
| , m_BackendProfilingContext(backendProfilingContexts) |
| {} |
| |
| void operator()(const arm::pipe::Packet& packet) override; |
| |
| void setTimelineEnabled(bool timelineEnabled) |
| { |
| m_TimelineEnabled = timelineEnabled; |
| } |
| |
| private: |
| const ICounterDirectory& m_CounterDirectory; |
| ISendCounterPacket& m_SendCounterPacket; |
| ISendTimelinePacket& m_SendTimelinePacket; |
| ProfilingStateMachine& m_StateMachine; |
| IProfilingServiceStatus& m_ProfilingServiceStatus; |
| Optional<BackendProfilingContexts> m_BackendProfilingContext; |
| bool m_TimelineEnabled = false; |
| }; |
| |
| } // namespace profiling |
| |
| } // namespace armnn |
| |