From 397043fa3d900430e9e0f6d328b76898f9613388 Mon Sep 17 00:00:00 2001 From: Jim Flynn Date: Thu, 17 Oct 2019 17:37:10 +0100 Subject: IVGCVSW-4002 Add FamilyId to CommandHandlerKey Change-Id: I0bb0bf77da2bcd7f4746078c4ccee9acc98638a7 Signed-off-by: Jim Flynn --- src/profiling/CommandHandlerRegistry.cpp | 13 ++++++++----- 1 file changed, 8 insertions(+), 5 deletions(-) (limited to 'src/profiling/CommandHandlerRegistry.cpp') diff --git a/src/profiling/CommandHandlerRegistry.cpp b/src/profiling/CommandHandlerRegistry.cpp index bd9b318835..8070afe623 100644 --- a/src/profiling/CommandHandlerRegistry.cpp +++ b/src/profiling/CommandHandlerRegistry.cpp @@ -14,11 +14,14 @@ namespace armnn namespace profiling { -void CommandHandlerRegistry::RegisterFunctor(CommandHandlerFunctor* functor, uint32_t packetId, uint32_t version) +void CommandHandlerRegistry::RegisterFunctor(CommandHandlerFunctor* functor, + uint32_t familyId, + uint32_t packetId, + uint32_t version) { BOOST_ASSERT_MSG(functor, "Provided functor should not be a nullptr"); - CommandHandlerKey key(packetId, version); + CommandHandlerKey key(familyId, packetId, version); registry[key] = functor; } @@ -26,12 +29,12 @@ void CommandHandlerRegistry::RegisterFunctor(CommandHandlerFunctor* functor) { BOOST_ASSERT_MSG(functor, "Provided functor should not be a nullptr"); - RegisterFunctor(functor, functor->GetPacketId(), functor->GetVersion()); + RegisterFunctor(functor, functor->GetFamilyId(), functor->GetPacketId(), functor->GetVersion()); } -CommandHandlerFunctor* CommandHandlerRegistry::GetFunctor(uint32_t packetId, uint32_t version) const +CommandHandlerFunctor* CommandHandlerRegistry::GetFunctor(uint32_t familyId,uint32_t packetId, uint32_t version) const { - CommandHandlerKey key(packetId, version); + CommandHandlerKey key(familyId, packetId, version); // Check that the requested key exists if (registry.find(key) == registry.end()) -- cgit v1.2.1