github/kubernetes
1
0
Fork 0
mirror of https://github.com/k3s-io/kubernetes.git synced 2025-08-09 03:57:41 +00:00
Code Issues Projects Releases Wiki Activity

Update test/e2e/scheduling/nvidia-gpus to also run cuda10 vector add.

Browse Source
This commit is contained in:
Jiaying Zhang 2019-03-21 16:29:47 -07:00
parent 0c93929298
commit 54c2c2690c
1 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
test/e2e/scheduling/nvidia-gpus.go
View File

@ -53,7 +53,7 @@ func makeCudaAdditionDevicePluginTestPod() *v1.Pod {
RestartPolicy: v1.RestartPolicyNever, RestartPolicy: v1.RestartPolicyNever,
Containers: []v1.Container{ Containers: []v1.Container{
{ {
Name: "vector-addition", Name: "vector-addition-cuda8",
Image: imageutils.GetE2EImage(imageutils.CudaVectorAdd), Image: imageutils.GetE2EImage(imageutils.CudaVectorAdd),
Resources: v1.ResourceRequirements{ Resources: v1.ResourceRequirements{
Limits: v1.ResourceList{ Limits: v1.ResourceList{
@ -61,6 +61,15 @@ func makeCudaAdditionDevicePluginTestPod() *v1.Pod {
}, },
}, },
}, },
{
Name: "vector-addition-cuda10",
Image: imageutils.GetE2EImage(imageutils.CudaVectorAdd2),
Resources: v1.ResourceRequirements{
Limits: v1.ResourceList{
gpuResourceName: *resource.NewQuantity(1, resource.DecimalSI),
},
},
},
}, },
}, },
} }