From abdc17bdcca5c2434bdf66937873f3caee812ee9 Mon Sep 17 00:00:00 2001
From: Matthew Douglas <38992547+matthewdouglas@users.noreply.github.com>
Date: Tue, 17 Feb 2026 10:13:23 -0500
Subject: [PATCH 1/2] CI: add NVIDIA A10 runner for tests

---
 .github/workflows/test-runner.yml   | 7 +++++--
 .github/workflows/tests-nightly.yml | 2 +-
 .github/workflows/tests-pr.yml      | 2 +-
 3 files changed, 7 insertions(+), 4 deletions(-)

diff --git a/.github/workflows/test-runner.yml b/.github/workflows/test-runner.yml
index 12c32b828..6783bf065 100644
--- a/.github/workflows/test-runner.yml
+++ b/.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
diff --git a/.github/workflows/tests-nightly.yml b/.github/workflows/tests-nightly.yml
index 60ed4fe3f..754b0cd36 100644
--- a/.github/workflows/tests-nightly.yml
+++ b/.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:
diff --git a/.github/workflows/tests-pr.yml b/.github/workflows/tests-pr.yml
index 7766089f6..ade85bcea 100644
--- a/.github/workflows/tests-pr.yml
+++ b/.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:

From 6294b717022faa4879b619c76d52a45b10eef400 Mon Sep 17 00:00:00 2001
From: Matthew Douglas <38992547+matthewdouglas@users.noreply.github.com>
Date: Fri, 20 Feb 2026 10:35:33 -0500
Subject: [PATCH 2/2] CI: build for sm80 when running tests

---
 .github/workflows/test-runner.yml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/.github/workflows/test-runner.yml b/.github/workflows/test-runner.yml
index 6783bf065..ad5415a8c 100644
--- a/.github/workflows/test-runner.yml
+++ b/.github/workflows/test-runner.yml
@@ -167,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