1 //
2 // Copyright © 2020 Arm Ltd. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5 #include "CounterIdMap.hpp"
6 #include "armnn/BackendId.hpp"
7 #include <armnn/Exceptions.hpp>
8 #include <map>
9
10 namespace armnn
11 {
12 namespace profiling
13 {
14
RegisterMapping(uint16_t globalCounterId,uint16_t backendCounterId,const armnn::BackendId & backendId)15 void CounterIdMap::RegisterMapping(uint16_t globalCounterId,
16 uint16_t backendCounterId,
17 const armnn::BackendId& backendId)
18 {
19 std::pair<uint16_t, armnn::BackendId> backendIdPair(backendCounterId, backendId);
20 m_GlobalCounterIdMap[globalCounterId] = backendIdPair;
21 m_BackendCounterIdMap[backendIdPair] = globalCounterId;
22 }
23
Reset()24 void CounterIdMap::Reset()
25 {
26 m_GlobalCounterIdMap.clear();
27 m_BackendCounterIdMap.clear();
28 }
29
GetGlobalId(uint16_t backendCounterId,const armnn::BackendId & backendId) const30 uint16_t CounterIdMap::GetGlobalId(uint16_t backendCounterId, const armnn::BackendId& backendId) const
31 {
32 std::pair<uint16_t, armnn::BackendId> backendIdPair(backendCounterId, backendId);
33 auto it = m_BackendCounterIdMap.find(backendIdPair);
34 if (it == m_BackendCounterIdMap.end())
35 {
36 std::stringstream ss;
37 ss << "No Backend Counter [" << backendIdPair.second << ":" << backendIdPair.first << "] registered";
38 throw armnn::Exception(ss.str());
39 }
40 return it->second;
41 }
42
GetBackendId(uint16_t globalCounterId) const43 const std::pair<uint16_t, armnn::BackendId>& CounterIdMap::GetBackendId(uint16_t globalCounterId) const
44 {
45 auto it = m_GlobalCounterIdMap.find(globalCounterId);
46 if (it == m_GlobalCounterIdMap.end())
47 {
48 std::stringstream ss;
49 ss << "No Global Counter ID [" << globalCounterId << "] registered";
50 throw armnn::Exception(ss.str());
51 }
52 return it->second;
53 }
54
55 } // namespace profiling
56 } // namespace armnn
57