From 1627c41f8404e8191e77a64e8c40a4a0382006c8 Mon Sep 17 00:00:00 2001 From: henrygd Date: Fri, 27 Jun 2025 18:00:47 -0400 Subject: [PATCH] fix gpu name issue introduced in previous commit --- beszel/internal/agent/gpu.go | 2 +- beszel/internal/agent/gpu_test.go | 10 ++++++++++ 2 files changed, 11 insertions(+), 1 deletion(-) diff --git a/beszel/internal/agent/gpu.go b/beszel/internal/agent/gpu.go index 5e274e8..7bf1091 100644 --- a/beszel/internal/agent/gpu.go +++ b/beszel/internal/agent/gpu.go @@ -243,7 +243,7 @@ func (gm *GPUManager) GetCurrentData() map[string]system.GPUData { // copy / reset the data gpuData := make(map[string]system.GPUData, len(gm.GpuDataMap)) for id, gpu := range gm.GpuDataMap { - var gpuAvg system.GPUData + gpuAvg := *gpu gpuAvg.Temperature = twoDecimals(gpu.Temperature) gpuAvg.MemoryUsed = twoDecimals(gpu.MemoryUsed) diff --git a/beszel/internal/agent/gpu_test.go b/beszel/internal/agent/gpu_test.go index 690be50..7d64da4 100644 --- a/beszel/internal/agent/gpu_test.go +++ b/beszel/internal/agent/gpu_test.go @@ -352,6 +352,15 @@ func TestGetCurrentData(t *testing.T) { Power: 60, Count: 1, }, + "2": { + Name: "GPU 2", + Temperature: 70, + MemoryUsed: 4096, + MemoryTotal: 8192, + Usage: 200, + Power: 400, + Count: 1, + }, }, } @@ -360,6 +369,7 @@ func TestGetCurrentData(t *testing.T) { // Verify name disambiguation assert.Equal(t, "GPU1 0", result["0"].Name) assert.Equal(t, "GPU1 1", result["1"].Name) + assert.Equal(t, "GPU 2", result["2"].Name) // Check averaged values in the result assert.InDelta(t, 50.0, result["0"].Usage, 0.01)