diff options
Diffstat (limited to 'src/profiling/RegisterBackendCounters.hpp')
-rw-r--r-- | src/profiling/RegisterBackendCounters.hpp | 62 |
1 files changed, 62 insertions, 0 deletions
diff --git a/src/profiling/RegisterBackendCounters.hpp b/src/profiling/RegisterBackendCounters.hpp new file mode 100644 index 0000000000..41886c0444 --- /dev/null +++ b/src/profiling/RegisterBackendCounters.hpp @@ -0,0 +1,62 @@ +// +// Copyright © 2020 Arm Ltd. All rights reserved. +// SPDX-License-Identifier: MIT +// + +#pragma once + +#include "armnn/backends/profiling/IBackendProfiling.hpp" +#include "CounterIdMap.hpp" +#include "CounterDirectory.hpp" +#include "ProfilingService.hpp" + +namespace armnn +{ + +namespace profiling +{ + +class RegisterBackendCounters : public IRegisterBackendCounters +{ +public: + + RegisterBackendCounters(uint16_t currentMaxGlobalCounterID, const BackendId& backendId) + : m_CurrentMaxGlobalCounterID(currentMaxGlobalCounterID), + m_CounterDirectory(ProfilingService::Instance().GetCounterRegistry()), + m_BackendId(backendId) {} + + ~RegisterBackendCounters() = default; + + void RegisterCategory(const std::string& categoryName, + const Optional<uint16_t>& deviceUid = EmptyOptional(), + const Optional<uint16_t>& counterSetUid = EmptyOptional()) override; + + uint16_t RegisterDevice(const std::string& deviceName, + uint16_t cores = 0, + const Optional<std::string>& parentCategoryName = EmptyOptional()) override; + + uint16_t RegisterCounterSet(const std::string& counterSetName, + uint16_t count = 0, + const Optional<std::string>& parentCategoryName = EmptyOptional()) override; + + uint16_t RegisterCounter(const uint16_t uid, + const std::string& parentCategoryName, + uint16_t counterClass, + uint16_t interpolation, + double multiplier, + const std::string& name, + const std::string& description, + const Optional<std::string>& units = EmptyOptional(), + const Optional<uint16_t>& numberOfCores = EmptyOptional(), + const Optional<uint16_t>& deviceUid = EmptyOptional(), + const Optional<uint16_t>& counterSetUid = EmptyOptional()) override; + +private: + uint16_t m_CurrentMaxGlobalCounterID; + ICounterRegistry& m_CounterDirectory; + const BackendId& m_BackendId; +}; + +} // namespace profiling + +} // namespace armnn
\ No newline at end of file |