From e6a2ccd09060ba93203ddc5a7f79260cedf2c147 Mon Sep 17 00:00:00 2001 From: Finn Williams Date: Thu, 27 Feb 2020 16:21:41 +0000 Subject: IVGCVSW-4161 Provide for per model call back registration !armnn:2810 Signed-off-by: Finn Williams Change-Id: Idf56d42bd767baa5df0059a2f489f75281f8ac71 --- src/timelineDecoder/TimelineCaptureCommandHandler.hpp | 17 +++++++++-------- 1 file changed, 9 insertions(+), 8 deletions(-) (limited to 'src/timelineDecoder/TimelineCaptureCommandHandler.hpp') diff --git a/src/timelineDecoder/TimelineCaptureCommandHandler.hpp b/src/timelineDecoder/TimelineCaptureCommandHandler.hpp index b69e615b56..e143b5f6e5 100644 --- a/src/timelineDecoder/TimelineCaptureCommandHandler.hpp +++ b/src/timelineDecoder/TimelineCaptureCommandHandler.hpp @@ -5,9 +5,9 @@ #pragma once -#include #include "armnn/profiling/ITimelineDecoder.hpp" +#include #include #include @@ -31,11 +31,11 @@ public: uint32_t packetId, uint32_t version, ITimelineDecoder& timelineDecoder, - uint32_t threadId_size) - : CommandHandlerFunctor(familyId, packetId, version), - m_TimelineDecoder(timelineDecoder), - m_ThreadIdSize(threadId_size), - m_PacketLength(0) + uint32_t threadIdSize = 0) + : CommandHandlerFunctor(familyId, packetId, version) + , m_TimelineDecoder(timelineDecoder) + , m_ThreadIdSize(threadIdSize) + , m_PacketLength(0) {} void operator()(const armnn::profiling::Packet& packet) override; @@ -46,12 +46,13 @@ public: void ReadRelationship(const unsigned char* data, uint32_t& offset); void ReadEvent(const unsigned char* data, uint32_t& offset); + void SetThreadIdSize(uint32_t size); + private: void ParseData(const armnn::profiling::Packet& packet); ITimelineDecoder& m_TimelineDecoder; - - const uint32_t m_ThreadIdSize; + uint32_t m_ThreadIdSize; unsigned int m_PacketLength; static const ReadFunction m_ReadFunctions[]; -- cgit v1.2.1