pmu-metrics/metrics.cpp at main · UWHustle/pmu-metrics · GitHub

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
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
#include "metrics.h"

// // Populated in perf_events.h
// std::map<int, std::string> event_names;

unsigned long _rdpmc(unsigned long pmc_id) {
#if _COLLECT_PMU_METRICS_
    unsigned long a, d;
    __asm__ __volatile__("rdpmc" : "=a" (a), "=d" (d) : "c" (pmc_id));
    return ((unsigned long)a) | (((unsigned long)d) << 32);
#else
    return 0;
#endif
}

unsigned long getTimeDiff(struct timespec start_time, struct timespec end_time) {
    return (unsigned long)((end_time.tv_sec - start_time.tv_sec)*1000000000 +
        double(end_time.tv_nsec - start_time.tv_nsec));
    return 0;
}

unsigned long * Metrics::pmu_ids = NULL;
unsigned long * Metrics::event_ids = NULL;

std::map<unsigned long, std::string> event_names_map_copy(event_names_map, event_names_map + num_events);

Metrics::Metrics() {
    n =  3 * FIXED_CTRS_ENBL + N_PERFEVTSEL_MSR;
    metrics = (unsigned long*)malloc(sizeof(unsigned long)*n);
    if (!pmu_ids) {
        // Initialize PMU ids
        pmu_ids = (unsigned long*)malloc(sizeof(unsigned long)*n);
        int idx = 0;
        if (FIXED_CTRS_ENBL) {
            pmu_ids[0] = (1<<30);
            pmu_ids[1] = (1<<30) + 1;
            pmu_ids[2] = (1<<30) + 2;
            idx = 3;
        }
        for (int i=0; i<N_PERFEVTSEL_MSR; i++,idx++) {
            pmu_ids[idx] = i;
        }

        // Initialize Event IDs
        event_ids = (unsigned long*)malloc(sizeof(unsigned long)*n);
        idx = 0;
        if (FIXED_CTRS_ENBL) {
            event_ids[0] = ARCH_INST_RETD;
            event_ids[1] = ARCH_CORE_CYCLES;
            event_ids[2] = ARCH_TSC_REF_CYCLES;
            idx = 3;
        }
#if PERFEVTSEL0
        event_ids[idx++] = PERFEVTSEL0;
#endif
#if PERFEVTSEL1
        event_ids[idx++] = PERFEVTSEL1;
#endif
#if PERFEVTSEL2
        event_ids[idx++] = PERFEVTSEL2;
#endif
#if PERFEVTSEL3
        event_ids[idx++] = PERFEVTSEL3;
#endif
#if PERFEVTSEL4
        event_ids[idx++] = PERFEVTSEL4;
#endif
#if PERFEVTSEL5
        event_ids[idx++] = PERFEVTSEL5;
#endif
#if PERFEVTSEL6
        event_ids[idx++] = PERFEVTSEL6;
#endif
#if PERFEVTSEL7
        event_ids[idx++] = PERFEVTSEL7;
#endif
    }
}

void getMetricsStart(Metrics &m) {
    clock_gettime(CLOCK_MONOTONIC, &m.startTime);
    for (int i=0; i<m.n; i++) {
        m.metrics[i] = _rdpmc(m.pmu_ids[i]);
    }
}

void getMetricsEnd(Metrics &m) {
    for (int i=0; i<m.n; i++) {
        m.metrics[i] = _rdpmc(m.pmu_ids[i]) - m.metrics[i];
    }
    clock_gettime(CLOCK_MONOTONIC, &m.endTime);
    m.timeElapsedns = getTimeDiff(m.startTime, m.endTime);
}

void printMetrics(Metrics &m) {
    unsigned long event_id;
    std::string event_name;
    std::cout << "Total time elapsed (ns): "
        << m.timeElapsedns
        << std::endl;
    std::cout << "--" << std::endl;
    std::cout
        << std::left
        << std::setw(40)
        << "Performance Event"
        << std::left
        << std::setw(25)
        << "Count"
        << std::endl;
    for (int i=0; i<m.n; i++) {
        event_id = m.event_ids[i];
        event_name = event_names_map_copy[event_id];
        std::cout
            << std::left
            << std::setw(40)
            << event_name
            << std::left
            << std::setw(25)
            << m.metrics[i]
            << std::endl;
    }
}