From ddd7c555769ac2064bde7a6f57c0620985b5bfc8 Mon Sep 17 00:00:00 2001
From: Wing Lian <wing@axolotl.ai>
Date: Fri, 16 May 2025 17:29:57 -0400
Subject: [PATCH] build hopper w fa3 on torch 2.6

---
 .github/workflows/base.yml | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/.github/workflows/base.yml b/.github/workflows/base.yml
index a28b7664a..c22dfa60d 100644
--- a/.github/workflows/base.yml
+++ b/.github/workflows/base.yml
@@ -52,11 +52,11 @@ jobs:
             python_version: "3.11"
             pytorch: 2.7.0
             torch_cuda_arch_list: "7.0 7.5 8.0 8.6 8.7 8.9 9.0+PTX"
-          - cuda: "128"
-            cuda_version: 12.8.1
+          - cuda: "126"
+            cuda_version: 12.6.3
             cudnn_version: ""
             python_version: "3.11"
-            pytorch: 2.7.0
+            pytorch: 2.6.0
             suffix: "-hopper"
             torch_cuda_arch_list: "9.0+PTX"
           - cuda: "128"