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

2019-03-21 16:29:47 -07:00
parent 0c93929298
commit 54c2c2690c
1 changed files with 10 additions and 1 deletions
--- a/test/e2e/scheduling/nvidia-gpus.go
+++ b/test/e2e/scheduling/nvidia-gpus.go
@@ -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),
+						},
+					},
+				},
 			},
 		},
 	}