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

Update test/e2e/scheduling/nvidia-gpus to also run cuda10 vector add.
k3s-v1.15.3
Kubernetes Prow Robot 2019-04-24 14:20:47 -07:00 committed by GitHub
commit 6cd85298c5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 10 additions and 1 deletions

View File

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