diff options
Diffstat (limited to 'lib/ethosu_monitor/include/ethosu_monitor.hpp')
-rw-r--r-- | lib/ethosu_monitor/include/ethosu_monitor.hpp | 66 |
1 files changed, 66 insertions, 0 deletions
diff --git a/lib/ethosu_monitor/include/ethosu_monitor.hpp b/lib/ethosu_monitor/include/ethosu_monitor.hpp new file mode 100644 index 0000000..c1308a5 --- /dev/null +++ b/lib/ethosu_monitor/include/ethosu_monitor.hpp @@ -0,0 +1,66 @@ +/* + * Copyright (c) 2021 Arm Limited. All rights reserved. + * + * SPDX-License-Identifier: Apache-2.0 + * + * Licensed under the Apache License, Version 2.0 (the License); you may + * not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an AS IS BASIS, WITHOUT + * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ + +#ifndef ETHOSU_MONITOR_H +#define ETHOSU_MONITOR_H + +#include "EventRecorder.h" +#include "EventRecorderConf.h" +#include <algorithm> +#include <ethosu_driver.h> +#include <pmu_ethosu.h> +#include <stdint.h> + +class EthosUMonitor { + +public: + enum Backend { PRINTF, EVENT_RECORDER }; + + EthosUMonitor(int32_t __id = EventID(EventLevelError, EvtStatistics_No, EventRecordNone), Backend backend = PRINTF); + + template <typename T> + void configure(ethosu_driver *drv, const T &eventIds) { + // Set event ids + numEvents = std::min(static_cast<size_t>(ETHOSU_PMU_NCOUNTERS), eventIds.size()); + for (size_t i = 0; i < numEvents; i++) { + ethosuEventIds[i] = eventIds[i]; + } + + // config pmu and set driver + ETHOSU_PMU_Enable(drv); + + for (size_t i = 0; i < ETHOSU_PMU_NCOUNTERS; i++) { + ETHOSU_PMU_Set_EVTYPER(drv, i, static_cast<ethosu_pmu_event_type>(ethosuEventIds[i])); + ETHOSU_PMU_CNTR_Enable(drv, 1 << i); + } + + ETHOSU_PMU_EVCNTR_ALL_Reset(drv); + } + + void release(ethosu_driver *drv); + + void monitorSample(ethosu_driver *drv); + +private: + ethosu_pmu_event_type ethosuEventIds[ETHOSU_PMU_NCOUNTERS]; + size_t numEvents; + int32_t eventRecordId; + Backend backend; +}; + +#endif |