From 98e18d59d2e9c0ecb1a509c1f8e60cd29049a57b Mon Sep 17 00:00:00 2001
From: Wing Lian <wing@axolotl.ai>
Date: Thu, 23 Apr 2026 15:05:01 -0400
Subject: [PATCH] larger torch cuda arch list coverage for older cuda

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

diff --git a/.github/workflows/base.yml b/.github/workflows/base.yml
index 946e8fd61..5062545c1 100644
--- a/.github/workflows/base.yml
+++ b/.github/workflows/base.yml
@@ -229,7 +229,7 @@ jobs:
             cudnn_version: ""
             python_version: "3.12"
             pytorch: 2.11.0
-            torch_cuda_arch_list: "9.0+PTX"
+            torch_cuda_arch_list: "7.0 7.5 8.0 8.6 8.7 8.9 9.0+PTX"
             dockerfile: "Dockerfile-uv-base"
             platforms: "linux/amd64,linux/arm64"
           - cuda: "130"