add linux-aarch64 GPU runner

NVIDIA · Dec 12, 2024 · 535cedd · 535cedd
1 parent 64cbc4c
commit 535cedd
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/.github/workflows/gh-build-and-test.yml b/.github/workflows/gh-build-and-test.yml
@@ -82,7 +82,8 @@ jobs:
     permissions:
       id-token: write # This is required for configure-aws-credentials
       contents: read  # This is required for actions/checkout
-    runs-on: ${{ (inputs.host-platform == 'linux-x64' && 'linux-amd64-gpu-v100-latest-1') }}
+    runs-on: ${{ (inputs.host-platform == 'linux-x64' && 'linux-amd64-gpu-v100-latest-1') ||
+                 (inputs.host-platform == 'linux-aarch64' && 'linux-arm64-gpu-a100-latest-1') }}
     # TODO: use a different (nvidia?) container, or just run on bare image
     container:
       options: -u root --security-opt seccomp=unconfined --privileged --shm-size 16g