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
|
/*
* Copyright (C) 2020-2022 Intel Corporation
*
* SPDX-License-Identifier: MIT
*
*/
#pragma once
#include "level_zero/tools/source/metrics/metric.h"
struct Event;
namespace L0 {
struct OaMetricStreamerImp : MetricStreamer {
~OaMetricStreamerImp() override{};
ze_result_t readData(uint32_t maxReportCount, size_t *pRawDataSize, uint8_t *pRawData) override;
ze_result_t close() override;
ze_result_t initialize(ze_device_handle_t hDevice, zet_metric_group_handle_t hMetricGroup);
ze_result_t startMeasurements(uint32_t ¬ifyEveryNReports, uint32_t &samplingPeriodNs);
Event::State getNotificationState() override;
void attachEvent(ze_event_handle_t hNotificationEvent);
void detachEvent();
ze_result_t appendStreamerMarker(CommandList &commandList, uint32_t value) override;
std::vector<zet_metric_streamer_handle_t> &getMetricStreamers();
protected:
ze_result_t stopMeasurements();
uint32_t getOaBufferSize(const uint32_t notifyEveryNReports) const;
uint32_t getNotifyEveryNReports(const uint32_t oaBufferSize) const;
uint32_t getRequiredBufferSize(const uint32_t maxReportCount) const;
ze_device_handle_t hDevice = nullptr;
zet_metric_group_handle_t hMetricGroup = nullptr;
Event *pNotificationEvent = nullptr;
uint32_t rawReportSize = 0;
uint32_t oaBufferSize = 0;
std::vector<zet_metric_streamer_handle_t> metricStreamers;
};
} // namespace L0
|
