diff --git a/providers/gpgpu/debian/control b/providers/gpgpu/debian/control index 6e4e3c25b9..50e2138c0b 100644 --- a/providers/gpgpu/debian/control +++ b/providers/gpgpu/debian/control @@ -16,7 +16,7 @@ Build-Depends: checkbox-ng, Standards-Version: 3.9.6 Package: checkbox-provider-gpgpu -Architecture: amd64 +Architecture: any-amd64 arm64 Depends: checkbox-provider-base, ${plainbox:Depends}, ${misc:Depends} Suggests: ubuntu-drivers-common, ${plainbox:Suggests} X-Plainbox-Provider: yes @@ -24,7 +24,7 @@ Description: Checkbox provider for GPGPU testing This package provides a test plan and tooling to be used by Canonical for the testing and certification of GPGPU devices in "server" computer systems . - This provider depends heavily on the installation of the latest nVidia drivers - and the CUDA tool kit from nVidia, as well as testing software hosted on - github. + This provider depends heavily on the installation of the latest NVIDIA drivers + and the CUDA tool kit from NVIDIA, as well as testing software hosted on + GitHub. diff --git a/providers/gpgpu/debian/postinst b/providers/gpgpu/debian/postinst index 8c83962d6e..1cef30ebbc 100644 --- a/providers/gpgpu/debian/postinst +++ b/providers/gpgpu/debian/postinst @@ -1,3 +1,9 @@ #!/usr/bin/env bash -snap install gpu-burn rocm-validation-suite cuda-samples +machine=$(uname -m) + +if [[ "$machine" = "x86_64" ]]; then + snap install rocm-validation-suite +fi + +snap install lxd gpu-burn cuda-samples diff --git a/providers/gpgpu/units/jobs.pxu b/providers/gpgpu/units/jobs.pxu index 75e554ad2d..b826173d63 100644 --- a/providers/gpgpu/units/jobs.pxu +++ b/providers/gpgpu/units/jobs.pxu @@ -5,7 +5,7 @@ estimated_duration: 14400 requires: graphics_card.vendor == 'NVIDIA Corporation' snap.name == 'gpu-burn' - uname.machine == 'x86_64' + uname.machine in ['x86_64', 'aarch64'] _summary: NVIDIA GPGPU stress testing command: set -eo pipefail @@ -19,7 +19,7 @@ estimated_duration: 4 requires: graphics_card.vendor == 'NVIDIA Corporation' snap.name == 'cuda-samples' - uname.machine == 'x86_64' + uname.machine in ['x86_64', 'aarch64'] _summary: NVIDIA GPGPU query device test command: cuda-samples 1_Utilities deviceQueryDrv deviceQueryDrv _siblings: [