From e4a33ec3a0ba310dd05e6e07cec3229eed3ea780 Mon Sep 17 00:00:00 2001 From: Nimbus318 <136771156+Nimbus318@users.noreply.github.com> Date: Thu, 9 Jan 2025 20:02:48 +0800 Subject: [PATCH] chore: Adjust video memory calculation to match HAMi's Node annotation Signed-off-by: Nimbus318 <136771156+Nimbus318@users.noreply.github.com> --- server/internal/exporter/exporter.go | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/server/internal/exporter/exporter.go b/server/internal/exporter/exporter.go index 345ef68..755a11a 100644 --- a/server/internal/exporter/exporter.go +++ b/server/internal/exporter/exporter.go @@ -101,10 +101,10 @@ func (s *MetricsGenerator) GenerateDeviceMetrics(ctx context.Context) error { if err == nil { HamiMemoryUsed.WithLabelValues(device.NodeName, provider, device.Type, device.Id, driver, deviceNo).Set(float64(deviceMemUsed)) } + HamiMemorySize.WithLabelValues(device.NodeName, provider, device.Type, device.Id, driver, deviceNo).Set(float64(device.Devmem)) + HamiMemoryUtil.WithLabelValues(device.NodeName, provider, device.Type, device.Id, driver, deviceNo).Set(roundToOneDecimal(100 * float64(deviceMemUsed/float32(device.Devmem)))) deviceMemSize, err := s.deviceMemTotal(ctx, provider, device.Id) if err == nil && deviceMemSize > 0 { - HamiMemorySize.WithLabelValues(device.NodeName, provider, device.Type, device.Id, driver, deviceNo).Set(float64(deviceMemSize)) - HamiMemoryUtil.WithLabelValues(device.NodeName, provider, device.Type, device.Id, driver, deviceNo).Set(roundToOneDecimal(100 * float64(deviceMemUsed/deviceMemSize))) HamiVMemoryScaling.WithLabelValues(device.NodeName, provider, device.Type, device.Id, driver, deviceNo).Set(roundToOneDecimal(float64(float32(device.Devmem) / deviceMemSize))) } actualCoreUtil, err := s.deviceCoreUtil(ctx, provider, device.Id)