add torch 2.9.0 to ci (#3223)

2025-10-30 18:50:26 -04:00
parent 4b1b4fa6d8
commit 633afffacb
3 changed files with 27 additions and 13 deletions
--- a/.github/workflows/multi-gpu-e2e.yml
+++ b/.github/workflows/multi-gpu-e2e.yml
@@ -40,6 +40,13 @@ jobs:
            axolotl_extras: fbgemm-gpu
            num_gpus: 2
            nightly_build: "true"
+          - cuda: 128
+            cuda_version: 12.8.1
+            python_version: "3.11"
+            pytorch: 2.9.0
+            axolotl_extras: fbgemm-gpu
+            num_gpus: 2
+            nightly_build: "true"
    runs-on: [self-hosted, modal]
    timeout-minutes: 120
    steps: