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
|
// SPDX-License-Identifier: GPL-3.0-or-later
package intelgpu
import (
"encoding/json"
"errors"
"fmt"
)
type (
gpuSummaryStats struct {
Frequency struct {
Actual float64 `json:"actual"`
} `json:"frequency"`
Power struct {
GPU float64 `json:"gpu"`
Package float64 `json:"package"`
} `json:"power"`
Engines map[string]struct {
Busy float64 `json:"busy"`
} `json:"engines"`
}
)
const precision = 100
func (ig *IntelGPU) collect() (map[string]int64, error) {
if ig.exec == nil {
return nil, errors.New("collector not initialized")
}
stats, err := ig.getGPUSummaryStats()
if err != nil {
return nil, err
}
mx := make(map[string]int64)
mx["frequency_actual"] = int64(stats.Frequency.Actual * precision)
mx["power_gpu"] = int64(stats.Power.GPU * precision)
mx["power_package"] = int64(stats.Power.Package * precision)
for name, es := range stats.Engines {
if !ig.engines[name] {
ig.addEngineCharts(name)
ig.engines[name] = true
}
key := fmt.Sprintf("engine_%s_busy", name)
mx[key] = int64(es.Busy * precision)
}
return mx, nil
}
func (ig *IntelGPU) getGPUSummaryStats() (*gpuSummaryStats, error) {
bs, err := ig.exec.queryGPUSummaryJson()
if err != nil {
return nil, err
}
if len(bs) == 0 {
return nil, errors.New("query returned empty response")
}
var stats gpuSummaryStats
if err := json.Unmarshal(bs, &stats); err != nil {
return nil, err
}
if len(stats.Engines) == 0 {
return nil, errors.New("query returned unexpected response")
}
return &stats, nil
}
|
