ClusterCockpit/cc-metric-collector
1
0
Fork 0
mirror of https://github.com/ClusterCockpit/cc-metric-collector.git synced 2025-11-04 10:45:06 +01:00
Code Issues Packages Projects Releases Wiki Activity

Option to use MIG UUID as subtype-id in NvidiaCollector

Browse Source
This commit is contained in:
Thomas Roehl
2022-05-13 14:34:32 +02:00
parent 826f364772
commit d4c89a4206
1 changed files with 23 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
collectors/nvidiaMetric.go
View File

@@ -21,6 +21,7 @@ type NvidiaCollectorConfig struct {
AddBoardNumberMeta bool `json:"add_board_number_meta,omitempty"`
AddSerialMeta bool `json:"add_serial_meta,omitempty"`
ProcessMigDevices bool `json:"process_mig_devices,omitempty"`
UseUuidForMigDevices bool `json:"use_uuid_for_mig_device,omitempty"`
}
type NvidiaCollectorDevice struct {
@@ -49,6 +50,8 @@ func (m *NvidiaCollector) Init(config json.RawMessage) error {
m.name = "NvidiaCollector"
m.config.AddPciInfoTag = false
m.config.UsePciInfoAsTypeId = false
m.config.ProcessMigDevices = false
m.config.UseUuidForMigDevices = false
m.setup()
if len(config) > 0 {
err = json.Unmarshal(config, &m.config)
@@ -1143,12 +1146,21 @@ func (m *NvidiaCollector) Read(interval time.Duration, output chan lp.CCMetric)
for k, v := range m.gpus[i].tags {
migDevice.tags[k] = v
}
m.gpus[i].tags["stype"] = "mig"
m.gpus[i].tags["stype-id"] = fmt.Sprintf("%d", j)
migDevice.tags["stype"] = "mig"
if !m.config.UseUuidForMigDevices {
migDevice.tags["stype-id"] = fmt.Sprintf("%d", j)
} else {
uuid, ret := nvml.DeviceGetUUID(mdev)
if ret != nvml.SUCCESS {
cclog.ComponentError(m.name, "Unable to get UUID for mig device at index", j, ":", err.Error())
} else {
migDevice.tags["stype-id"] = uuid
}
}
for k, v := range m.gpus[i].meta {
migDevice.meta[k] = v
}
if _, ok := migDevice.meta["uuid"]; ok {
if _, ok := migDevice.meta["uuid"]; ok && !m.config.UseUuidForMigDevices {
uuid, ret := nvml.DeviceGetUUID(mdev)
if ret == nvml.SUCCESS {
migDevice.meta["uuid"] = uuid