diff --git a/sdl/gpu.go b/sdl/gpu.go index 1bb6b1255..0fa466011 100644 --- a/sdl/gpu.go +++ b/sdl/gpu.go @@ -17,7 +17,7 @@ type v2GPUNvidia struct { func (sdl *v2GPUNvidia) String() string { key := sdl.Model if sdl.RAM != nil { - key += "/" + sdl.RAM.StringWithSuffix("Gi") + key += "/ram/" + sdl.RAM.StringWithSuffix("Gi") } return key diff --git a/sdl/gpu_test.go b/sdl/gpu_test.go index c409f0820..282239296 100644 --- a/sdl/gpu_test.go +++ b/sdl/gpu_test.go @@ -69,7 +69,7 @@ attributes: require.NoError(t, err) require.Equal(t, gpuQuantity(1), p.Units) require.Equal(t, 1, len(p.Attributes)) - require.Equal(t, "vendor/nvidia/model/a100/80Gi", p.Attributes[0].Key) + require.Equal(t, "vendor/nvidia/model/a100/ram/80Gi", p.Attributes[0].Key) require.Equal(t, "true", p.Attributes[0].Value) } @@ -105,9 +105,9 @@ attributes: require.NoError(t, err) require.Equal(t, gpuQuantity(1), p.Units) require.Equal(t, 2, len(p.Attributes)) - require.Equal(t, "vendor/nvidia/model/a100/40Gi", p.Attributes[0].Key) + require.Equal(t, "vendor/nvidia/model/a100/ram/40Gi", p.Attributes[0].Key) require.Equal(t, "true", p.Attributes[0].Value) - require.Equal(t, "vendor/nvidia/model/a100/80Gi", p.Attributes[1].Key) + require.Equal(t, "vendor/nvidia/model/a100/ram/80Gi", p.Attributes[1].Key) require.Equal(t, "true", p.Attributes[1].Value) } @@ -129,7 +129,7 @@ attributes: require.Equal(t, 2, len(p.Attributes)) require.Equal(t, "vendor/nvidia/model/a100", p.Attributes[0].Key) require.Equal(t, "true", p.Attributes[0].Value) - require.Equal(t, "vendor/nvidia/model/a100/80Gi", p.Attributes[1].Key) + require.Equal(t, "vendor/nvidia/model/a100/ram/80Gi", p.Attributes[1].Key) require.Equal(t, "true", p.Attributes[1].Value) }