diff --git a/.gitignore b/.gitignore index 08e658f73..0b87f1d36 100644 --- a/.gitignore +++ b/.gitignore @@ -8,4 +8,6 @@ libvgpu.so .idea vendor license -vgpuvalidator \ No newline at end of file +vgpuvalidator +_output/ +coverage.out \ No newline at end of file diff --git a/pkg/device/ascend/device_test.go b/pkg/device/ascend/device_test.go index 2f04934f9..81ea3a612 100644 --- a/pkg/device/ascend/device_test.go +++ b/pkg/device/ascend/device_test.go @@ -312,8 +312,8 @@ func Test_PatchAnnotations(t *testing.T) { }, }, want: map[string]string{ - util.InRequestDevices[dev.config.CommonWord]: "device-0,Ascend,8738,1:;", - util.SupportDevices[dev.config.CommonWord]: "device-0,Ascend,8738,1:;", + util.InRequestDevices[dev.config.CommonWord]: "device-0,Ascend,8738,1,0:;", + util.SupportDevices[dev.config.CommonWord]: "device-0,Ascend,8738,1,0:;", "predicate-time": strconv.FormatInt(time.Now().Unix(), 10), "huawei.com/Ascend910A": "[{\"UUID\":\"device-0\",\"temp\":\"vir08\"}]", }, diff --git a/pkg/device/iluvatar/device_test.go b/pkg/device/iluvatar/device_test.go index df98b652e..b75e5622a 100644 --- a/pkg/device/iluvatar/device_test.go +++ b/pkg/device/iluvatar/device_test.go @@ -129,8 +129,8 @@ func TestPatchAnnotations(t *testing.T) { }, }, expected: map[string]string{ - util.InRequestDevices[IluvatarGPUDevice]: "k8s-gpu-iluvatar-0,Iluvatar,0,0:;", - util.SupportDevices[IluvatarGPUDevice]: "k8s-gpu-iluvatar-0,Iluvatar,0,0:;", + util.InRequestDevices[IluvatarGPUDevice]: "k8s-gpu-iluvatar-0,Iluvatar,0,0,0:;", + util.SupportDevices[IluvatarGPUDevice]: "k8s-gpu-iluvatar-0,Iluvatar,0,0,0:;", "iluvatar.ai/gpu-assigned": "false", "iluvatar.ai/predicate-time": strconv.FormatInt(time.Now().UnixNano(), 10), IluvatarDeviceSelection + "0": "0", diff --git a/pkg/device/nvidia/device_test.go b/pkg/device/nvidia/device_test.go index 236432cb0..35b7fcced 100644 --- a/pkg/device/nvidia/device_test.go +++ b/pkg/device/nvidia/device_test.go @@ -504,8 +504,8 @@ func Test_PatchAnnotations(t *testing.T) { }, }, want: map[string]string{ - util.InRequestDevices[NvidiaGPUDevice]: "nvidia-device-0,NVIDIA,2000,1:;", - util.SupportDevices[NvidiaGPUDevice]: "nvidia-device-0,NVIDIA,2000,1:;", + util.InRequestDevices[NvidiaGPUDevice]: "nvidia-device-0,NVIDIA,2000,1,0:;", + util.SupportDevices[NvidiaGPUDevice]: "nvidia-device-0,NVIDIA,2000,1,0:;", }, }, {