Merge pull request #75566 from jiayingz/gpu-test-update

Update test/e2e/scheduling/nvidia-gpus to also run cuda10 vector add.
This commit is contained in:
Kubernetes Prow Robot
2019-04-24 14:20:47 -07:00
committed by GitHub

View File

@@ -54,7 +54,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{
@@ -62,6 +62,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),
},
},
},
}, },
}, },
} }