in src/worker/exporters/custom_exporter.py [0:0]
def update_gauges_field(self, field_name, value, config):
if "gpu_" in field_name:
gpu_idx = field_name.split('_')[1]
metric_name = f"{field_name.split('_')[0]}_{field_name.split('_')[2]}"
if int(gpu_idx) < len(GPU_Mapping):
self.gauges[metric_name].labels(
config['gpu_id'][gpu_idx],
config['job_id']
).set(value)
elif "ib_" in field_name:
ib_port = field_name.split('_')[1]
metric_name = f"{field_name.split('_')[0]}_{field_name.split('_')[2]}"
if int(ib_port) < len(IB_Mapping):
self.gauges[metric_name].labels(
config['ib_port'][ib_port],
config['job_id']
).set(value)
else:
self.gauges[field_name].labels(
config['job_id']
).set(value)