From cc174b115e711d03a80836419f4912488013c289 Mon Sep 17 00:00:00 2001 From: Evan Lezar Date: Wed, 28 Aug 2024 08:49:51 +0200 Subject: [PATCH] [no-relnote] Use nbody as test job Signed-off-by: Evan Lezar --- tests/e2e/common/gpu_job.go | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/e2e/common/gpu_job.go b/tests/e2e/common/gpu_job.go index ee73cb5f1..fb512f36f 100644 --- a/tests/e2e/common/gpu_job.go +++ b/tests/e2e/common/gpu_job.go @@ -38,7 +38,8 @@ var GPUJob = &batchv1.Job{ Containers: []v1.Container{ { Name: "cuda-container", - Image: "nvcr.io/nvidia/k8s/cuda-sample:vectoradd-cuda10.2", + Image: "nvcr.io/nvidia/k8s/cuda-sample:nbody-cuda11.7.1-ubuntu18.04", + Args: []string{"--benchmark", "--numbodies=10000"}, Resources: v1.ResourceRequirements{ Limits: v1.ResourceList{ "nvidia.com/gpu": resource.MustParse("1"),