blob: a366898377ba5e6fe275fc9d293369c04fb7395a (
plain)
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
|
//
// Copyright © 2017 Arm Ltd. All rights reserved.
// SPDX-License-Identifier: MIT
//
#include <armnn/BackendId.hpp>
#include "Holder.hpp"
namespace armnn
{
namespace profiling
{
CaptureData& CaptureData::operator=(const CaptureData& other)
{
m_CapturePeriod = other.m_CapturePeriod;
m_CounterIds = other.m_CounterIds;
m_ActiveBackends = other.m_ActiveBackends;
return *this;
}
void CaptureData::SetActiveBackends(const std::set<armnn::BackendId>& activeBackends)
{
m_ActiveBackends = activeBackends;
}
void CaptureData::SetCapturePeriod(uint32_t capturePeriod)
{
m_CapturePeriod = capturePeriod;
}
void CaptureData::SetCounterIds(const std::vector<uint16_t>& counterIds)
{
m_CounterIds = counterIds;
}
const std::set<armnn::BackendId>& CaptureData::GetActiveBackends() const
{
return m_ActiveBackends;
}
uint32_t CaptureData::GetCapturePeriod() const
{
return m_CapturePeriod;
}
const std::vector<uint16_t>& CaptureData::GetCounterIds() const
{
return m_CounterIds;
}
CaptureData Holder::GetCaptureData() const
{
std::lock_guard<std::mutex> lockGuard(m_CaptureThreadMutex);
return m_CaptureData;
}
bool CaptureData::IsCounterIdInCaptureData(uint16_t counterId)
{
for (auto m_CounterId : m_CounterIds) {
if (m_CounterId == counterId)
{
return true;
}
}
// Return false by default unless counterId is found
return false;
}
void Holder::SetCaptureData(uint32_t capturePeriod,
const std::vector<uint16_t>& counterIds,
const std::set<armnn::BackendId>& activeBackends)
{
std::lock_guard<std::mutex> lockGuard(m_CaptureThreadMutex);
m_CaptureData.SetCapturePeriod(capturePeriod);
m_CaptureData.SetCounterIds(counterIds);
m_CaptureData.SetActiveBackends(activeBackends);
}
} // namespace profiling
} // namespace armnn
|
