From 54c2c2690cb7e33bf20e4f15896aca96618852fc Mon Sep 17 00:00:00 2001 From: Jiaying Zhang Date: Thu, 21 Mar 2019 16:29:47 -0700 Subject: [PATCH] Update test/e2e/scheduling/nvidia-gpus to also run cuda10 vector add. --- test/e2e/scheduling/nvidia-gpus.go | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/test/e2e/scheduling/nvidia-gpus.go b/test/e2e/scheduling/nvidia-gpus.go index 8dc0a3873dd..c3aa9f0ea2c 100644 --- 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), + }, + }, + }, }, }, }