diff options
Diffstat (limited to 'src/profiling/CounterIdMap.cpp')
| -rw-r--r-- | src/profiling/CounterIdMap.cpp | 50 |
1 files changed, 50 insertions, 0 deletions
diff --git a/src/profiling/CounterIdMap.cpp b/src/profiling/CounterIdMap.cpp new file mode 100644 index 0000000000..a925fb9cb4 --- /dev/null +++ b/src/profiling/CounterIdMap.cpp @@ -0,0 +1,50 @@ +// +// Copyright © 2020 Arm Ltd. All rights reserved. +// SPDX-License-Identifier: MIT +// +#include "CounterIdMap.hpp" +#include "armnn/BackendId.hpp" +#include <map> +#include <armnn/Exceptions.hpp> + +namespace armnn +{ +namespace profiling +{ + +void CounterIdMap::RegisterMapping(uint16_t globalCounterId, + uint16_t backendCounterId, + const armnn::BackendId& backendId) +{ + std::pair<uint16_t, armnn::BackendId> backendIdPair(backendCounterId, backendId); + m_GlobalCounterIdMap[globalCounterId] = backendIdPair; + m_BackendCounterIdMap[backendIdPair] = globalCounterId; +} + +uint16_t CounterIdMap::GetGlobalId(uint16_t backendCounterId, const armnn::BackendId& backendId) +{ + std::pair<uint16_t, armnn::BackendId> backendIdPair(backendCounterId, backendId); + auto it = m_BackendCounterIdMap.find(backendIdPair); + if (it == m_BackendCounterIdMap.end()) + { + std::stringstream ss; + ss << "No Backend Counter [" << backendIdPair.second << ":" << backendIdPair.first << "] registered"; + throw armnn::Exception(ss.str()); + } + return it->second; +} + +const std::pair<uint16_t, armnn::BackendId>& CounterIdMap::GetBackendId(uint16_t globalCounterId) +{ + auto it = m_GlobalCounterIdMap.find(globalCounterId); + if (it == m_GlobalCounterIdMap.end()) + { + std::stringstream ss; + ss << "No Global Counter ID [" << globalCounterId << "] registered"; + throw armnn::Exception(ss.str()); + } + return it->second; +} + +} // namespace profiling +} // namespace armnn
\ No newline at end of file |