CI: add NVIDIA A10 runner for tests (#1874)

* CI: add NVIDIA A10 runner for tests

* CI: build for sm80 when running tests

Author: matthewdouglas

.github/workflows/test-runner.yml

@@ -26,7 +26,7 @@ on:
       gpu_type:
         type: string
         default: ""
-        description: "GPU type for CUDA testing: T4, L40S"
+        description: "GPU type for CUDA testing: T4, A10,L40S"
       # cpu_type currently only affects linux x64 CPU testing to select specific CPU architectures
       cpu_type:
         type: string
@@ -65,11 +65,14 @@ jobs:
                   T4)
                     TEST_RUNNER="bandb-aws-g4dn-4xlarge-plus-use1-public-80"
                     ;;
+                  A10)
+                    TEST_RUNNER="bandb-aws-g5-4xlarge-plus-use1-public-80"
+                    ;;
                   L40S)
                     TEST_RUNNER="bandb-aws-g6e-4xlarge-plus-use1-public-80"
                     ;;
                   *)
-                    echo "::error::Must specify gpu_type (T4 or L40S) for linux-x64 cuda backend"
+                    echo "::error::Must specify gpu_type (T4, A10, L40S) for linux-x64 cuda backend"
                     exit 1
                     ;;
                 esac
@@ -164,7 +167,7 @@ jobs:
         run: bash .github/scripts/build-cuda.sh
         env:
           cuda_version: ${{ inputs.cuda_version }}
-          cuda_targets: "75;89"
+          cuda_targets: "75;80;89"
 
       - name: Upload build artifact
         uses: actions/upload-artifact@v4

.github/workflows/tests-nightly.yml

@@ -56,7 +56,7 @@ jobs:
       matrix:
         # Linux x64 cross-product
         platform: [linux-x64]
-        gpu_type: [T4, L40S]
+        gpu_type: [T4, A10, L40S]
         cuda_version: ["11.8.0", "12.6.3", "12.8.1", "13.0.2"]
 
         include:

.github/workflows/tests-pr.yml

@@ -64,7 +64,7 @@ jobs:
       fail-fast: false
       matrix:
         platform: [linux-x64]
-        gpu_type: [T4, L40S]
+        gpu_type: [T4, A10, L40S]
         cuda_version: ["11.8.0", "12.8.1", "13.0.2"]
 
         include: