From 2aed699d8ca4e84051afdab7d626f916bb9f902a Mon Sep 17 00:00:00 2001 From: briwan01 Date: Mon, 9 Jan 2023 17:44:02 +0800 Subject: [PATCH] test images: Update CudaVectorAdd test image to support arm64 Signed-off-by: briwan01 --- test/images/cuda-vector-add/BASEIMAGE | 5 +++-- test/images/cuda-vector-add/Dockerfile | 2 +- test/images/cuda-vector-add/VERSION | 2 +- 3 files changed, 5 insertions(+), 4 deletions(-) diff --git a/test/images/cuda-vector-add/BASEIMAGE b/test/images/cuda-vector-add/BASEIMAGE index 5e5cd4dac28..a89f93aef66 100644 --- a/test/images/cuda-vector-add/BASEIMAGE +++ b/test/images/cuda-vector-add/BASEIMAGE @@ -1,2 +1,3 @@ -linux/amd64=nvidia/cuda:10.0-devel-ubuntu18.04 -linux/ppc64le=nvidia/cuda-ppc64le:10.0-devel-ubuntu18.04 +linux/amd64=nvidia/cuda:11.1.1-devel-ubuntu18.04 +linux/arm64=nvidia/cuda:11.1.1-devel-ubuntu18.04 +linux/ppc64le=nvidia/cuda:11.1.1-devel-ubuntu18.04 diff --git a/test/images/cuda-vector-add/Dockerfile b/test/images/cuda-vector-add/Dockerfile index af82e6a5e9b..c2c6d10a5f2 100644 --- a/test/images/cuda-vector-add/Dockerfile +++ b/test/images/cuda-vector-add/Dockerfile @@ -15,8 +15,8 @@ ARG BASEIMAGE FROM $BASEIMAGE +ENV CUDA_PKG_VERSION=11-1 CROSS_BUILD_COPY qemu-QEMUARCH-static /usr/bin/ - RUN apt-get update && apt-get install -y --no-install-recommends \ cuda-samples-$CUDA_PKG_VERSION && \ rm -rf /var/lib/apt/lists/* diff --git a/test/images/cuda-vector-add/VERSION b/test/images/cuda-vector-add/VERSION index 8bbe6cf74a1..bb576dbde10 100644 --- a/test/images/cuda-vector-add/VERSION +++ b/test/images/cuda-vector-add/VERSION @@ -1 +1 @@ -2.2 +2.3