1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
|
//
// Copyright © 2017 Arm Ltd. All rights reserved.
// SPDX-License-Identifier: MIT
//
#include "CounterDirectory.hpp"
#include <armnn/Exceptions.hpp>
namespace armnn
{
namespace profiling
{
CounterDirectory::CounterDirectory(uint16_t uid,
const std::string& name,
uint16_t deviceCount,
uint16_t counterCount,
uint16_t categoryCount)
: m_Uid(uid)
, m_Name(name)
, m_DeviceCount(deviceCount)
, m_CounterCount(counterCount)
, m_CategoryCount(categoryCount)
, m_DeviceIds(deviceCount)
, m_CounterIds(counterCount)
, m_CategoryIds(categoryCount)
, m_DeviceObjects(deviceCount)
, m_CounterObjects(counterCount)
, m_CategoryObjects(categoryCount)
{}
// Helper methods
void CounterDirectory::CheckDeviceIndex(uint16_t index) const
{
if (index >= m_DeviceCount)
{
throw InvalidArgumentException("Invalid device index");
}
}
void CounterDirectory::CheckCounterIndex(uint16_t index) const
{
if (index >= m_CounterCount)
{
throw InvalidArgumentException("Invalid counter index");
}
}
void CounterDirectory::CheckCategoryIndex(uint16_t index) const
{
if (index >= m_CategoryCount)
{
throw InvalidArgumentException("Invalid category index");
}
}
// Getters for basic attributes
uint16_t CounterDirectory::GetUid() const
{
return m_Uid;
}
const std::string& CounterDirectory::GetName() const
{
return m_Name;
}
// Getters for counts
uint16_t CounterDirectory::GetDeviceCount() const
{
return m_DeviceCount;
}
uint16_t CounterDirectory::GetCounterCount() const
{
return m_CounterCount;
}
uint16_t CounterDirectory::GetCategoryCount() const
{
return m_CategoryCount;
}
// Getters and setters for devices
void CounterDirectory::GetDeviceValue(uint16_t index, uint32_t& value) const
{
CheckDeviceIndex(index);
value = m_DeviceIds[index].load();
}
void CounterDirectory::SetDeviceValue(uint16_t index, uint32_t value)
{
CheckDeviceIndex(index);
m_DeviceIds[index].store(value);
}
void CounterDirectory::GetDeviceObject(uint16_t index, Device* device) const
{
CheckDeviceIndex(index);
device = m_DeviceObjects[index].load();
}
void CounterDirectory::SetDeviceObject(uint16_t index, Device* device)
{
CheckDeviceIndex(index);
m_DeviceObjects[index].store(device);
}
// Getters and setters for counters
void CounterDirectory::GetCounterValue(uint16_t index, uint32_t& value) const
{
CheckCounterIndex(index);
value = m_CounterIds[index].load();
}
void CounterDirectory::SetCounterValue(uint16_t index, uint32_t value)
{
CheckCounterIndex(index);
m_CounterIds[index].store(value);
}
void CounterDirectory::GetCounterObject(uint16_t index, Counter* counter) const
{
CheckCounterIndex(index);
counter = m_CounterObjects[index].load();
}
void CounterDirectory::SetCounterObject(uint16_t index, Counter* counter)
{
CheckCounterIndex(index);
m_CounterObjects[index].store(counter);
}
// Getters and setters for categories
void CounterDirectory::GetCategoryValue(uint16_t index, uint32_t& value) const
{
CheckCategoryIndex(index);
value = m_CategoryIds[index].load();
}
void CounterDirectory::SetCategoryValue(uint16_t index, uint32_t value)
{
CheckCategoryIndex(index);
m_CategoryIds[index].store(value);
}
void CounterDirectory::GetCategoryObject(uint16_t index, Category* category) const
{
CheckCategoryIndex(index);
category = m_CategoryObjects[index].load();
}
void CounterDirectory::SetCategoryObject(uint16_t index, Category* category)
{
CheckCategoryIndex(index);
m_CategoryObjects[index].store(category);
}
} // namespace profiling
} // namespace armnn
|
