blob: 51174a72a8302fae7b303c722084468ac040443d (
plain)
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
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
|
/*
* Copyright © 2019-2021 Collabora, Ltd.
* Author: Antonio Caggiano <antonio.caggiano@collabora.com>
* Author: Robert Beckett <bob.beckett@collabora.com>
* Author: Corentin Noël <corentin.noel@collabora.com>
*
* SPDX-License-Identifier: MIT
*/
#pragma once
#include "pps.h"
#include "pps_driver.h"
namespace pps
{
struct GpuIncrementalState {
bool was_cleared = true;
};
struct GpuDataSourceTraits : public perfetto::DefaultDataSourceTraits {
using IncrementalStateType = GpuIncrementalState;
};
class Driver;
/// @brief This datasource samples performance counters at a specified rate.
/// Once the data is available, it sends a protobuf packet to the perfetto service.
/// At the very beginning, it sends a description of the counters.
/// After that, it sends counter values using the IDs set in the description.
class GpuDataSource : public perfetto::DataSource<GpuDataSource, GpuDataSourceTraits>
{
public:
void OnSetup(const SetupArgs &args) override;
void OnStart(const StartArgs &args) override;
void OnStop(const StopArgs &args) override;
/// Blocks until the data source starts
static void wait_started();
/// @brief Perfetto trace callback
static void trace_callback(TraceContext ctx);
static void register_data_source(const std::string &driver_name);
void trace(TraceContext &ctx);
private:
State state = State::Stop;
/// Time between trace callbacks
std::chrono::nanoseconds time_to_sleep = std::chrono::nanoseconds(1000000);
/// Used to check whether the datasource is quick enough
std::chrono::nanoseconds time_to_trace;
/// Last CPU timestamp at which we correlated CPU/GPU timestamps
uint64_t last_correlation_timestamp = 0;
/// A data source supports one driver at a time, but if you need more
/// than one gpu datasource you can just run another producer
Driver *driver = nullptr;
/// CPU timestamp of packet sent with counter descriptors
uint64_t descriptor_timestamp = 0;
/// GPU timestamp of packet sent with counter descriptors
uint64_t descriptor_gpu_timestamp = 0;
/// Used to track the first available counters
bool got_first_counters = false;
};
} // namespace pps
|
