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

This commit is contained in:
Jiaying Zhang 2019-03-21 16:29:47 -07:00
parent 0c93929298
commit 54c2c2690c

View File

@ -53,7 +53,7 @@ func makeCudaAdditionDevicePluginTestPod() *v1.Pod {
RestartPolicy: v1.RestartPolicyNever,
Containers: []v1.Container{
{
Name: "vector-addition",
Name: "vector-addition-cuda8",
Image: imageutils.GetE2EImage(imageutils.CudaVectorAdd),
Resources: v1.ResourceRequirements{
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),
},
},
},
},
},
}