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

Update test/e2e/scheduling/nvidia-gpus to also run cuda10 vector add.
2019-04-24 14:20:47 -07:00
parent ce3ddcd5f6 54c2c2690c
commit 6cd85298c5
1 changed files with 10 additions and 1 deletions
--- a/test/e2e/scheduling/nvidia-gpus.go
+++ b/test/e2e/scheduling/nvidia-gpus.go
@@ -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),
+						},
+					},
+				},
 			},
 		},
 	}