mirror of
https://github.com/intel/compute-runtime.git
synced 2025-12-20 00:24:58 +08:00
This Patch adds support for collecting IP Metrics using StreamerOpen, StreamerClose and StreamerReadData Related-To: LOCI-2755 Related-To: LOCI-2756 Signed-off-by: Joshua Santosh Ranjan <joshua.santosh.ranjan@intel.com>
89 lines
3.7 KiB
C++
89 lines
3.7 KiB
C++
/*
|
|
* Copyright (C) 2022 Intel Corporation
|
|
*
|
|
* SPDX-License-Identifier: MIT
|
|
*
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include "level_zero/tools/source/metrics/metric.h"
|
|
#include "level_zero/tools/source/metrics/os_metric_ip_sampling.h"
|
|
|
|
namespace L0 {
|
|
|
|
struct IpSamplingMetricImp;
|
|
struct IpSamplingMetricGroupImp;
|
|
struct IpSamplingMetricStreamerImp;
|
|
|
|
class IpSamplingMetricSourceImp : public MetricSource {
|
|
|
|
public:
|
|
IpSamplingMetricSourceImp(const MetricDeviceContext &metricDeviceContext);
|
|
virtual ~IpSamplingMetricSourceImp() = default;
|
|
void enable() override;
|
|
bool isAvailable() override;
|
|
ze_result_t metricGroupGet(uint32_t *pCount, zet_metric_group_handle_t *phMetricGroups) override;
|
|
ze_result_t appendMetricMemoryBarrier(CommandList &commandList) override;
|
|
void setMetricOsInterface(std::unique_ptr<MetricIpSamplingOsInterface> &metricOsInterface);
|
|
static std::unique_ptr<IpSamplingMetricSourceImp> create(const MetricDeviceContext &metricDeviceContext);
|
|
MetricIpSamplingOsInterface *getMetricOsInterface() { return metricOsInterface.get(); }
|
|
IpSamplingMetricStreamerImp *pActiveStreamer = nullptr;
|
|
|
|
protected:
|
|
void cacheMetricGroup();
|
|
bool isEnabled = false;
|
|
|
|
const MetricDeviceContext &metricDeviceContext;
|
|
std::unique_ptr<MetricIpSamplingOsInterface> metricOsInterface = nullptr;
|
|
std::unique_ptr<IpSamplingMetricGroupImp> cachedMetricGroup = nullptr;
|
|
};
|
|
|
|
struct IpSamplingMetricGroupImp : public MetricGroup {
|
|
IpSamplingMetricGroupImp(std::vector<IpSamplingMetricImp> &metrics);
|
|
virtual ~IpSamplingMetricGroupImp() = default;
|
|
|
|
ze_result_t getProperties(zet_metric_group_properties_t *pProperties) override;
|
|
ze_result_t metricGet(uint32_t *pCount, zet_metric_handle_t *phMetrics) override;
|
|
ze_result_t calculateMetricValues(const zet_metric_group_calculation_type_t type, size_t rawDataSize,
|
|
const uint8_t *pRawData, uint32_t *pMetricValueCount,
|
|
zet_typed_value_t *pMetricValues) override;
|
|
ze_result_t calculateMetricValuesExp(const zet_metric_group_calculation_type_t type, size_t rawDataSize,
|
|
const uint8_t *pRawData, uint32_t *pSetCount,
|
|
uint32_t *pTotalMetricValueCount, uint32_t *pMetricCounts,
|
|
zet_typed_value_t *pMetricValues) override;
|
|
bool activate() override;
|
|
bool deactivate() override;
|
|
zet_metric_group_handle_t getMetricGroupForSubDevice(const uint32_t subDeviceIndex) override;
|
|
ze_result_t streamerOpen(
|
|
zet_context_handle_t hContext,
|
|
zet_device_handle_t hDevice,
|
|
zet_metric_streamer_desc_t *desc,
|
|
ze_event_handle_t hNotificationEvent,
|
|
zet_metric_streamer_handle_t *phMetricStreamer) override;
|
|
ze_result_t metricQueryPoolCreate(
|
|
zet_context_handle_t hContext,
|
|
zet_device_handle_t hDevice,
|
|
const zet_metric_query_pool_desc_t *desc,
|
|
zet_metric_query_pool_handle_t *phMetricQueryPool) override;
|
|
static std::unique_ptr<IpSamplingMetricGroupImp> create(std::vector<IpSamplingMetricImp> &ipSamplingMetrics);
|
|
|
|
private:
|
|
std::vector<std::unique_ptr<IpSamplingMetricImp>> metrics = {};
|
|
zet_metric_group_properties_t properties = {};
|
|
};
|
|
|
|
struct IpSamplingMetricImp : public Metric {
|
|
virtual ~IpSamplingMetricImp() = default;
|
|
IpSamplingMetricImp(zet_metric_properties_t &properties);
|
|
ze_result_t getProperties(zet_metric_properties_t *pProperties) override;
|
|
|
|
private:
|
|
zet_metric_properties_t properties;
|
|
};
|
|
|
|
template <>
|
|
IpSamplingMetricSourceImp &MetricDeviceContext::getMetricSource<IpSamplingMetricSourceImp>() const;
|
|
|
|
} // namespace L0
|