From 0d92c1093c96bf94fb37a03c7982454f4fb5672e Mon Sep 17 00:00:00 2001
From: Junya Morioka <mjun@mjunya.com>
Date: Thu, 16 Oct 2025 13:25:43 +0900
Subject: [PATCH] feat: add torch==2.9.0

---
 .github/workflows/build.yml | 26 ++++++++++++++++----------
 get_torch_cuda_version.py   |  2 +-
 2 files changed, 17 insertions(+), 11 deletions(-)

diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
index 379a08a..30adfaa 100644
--- a/.github/workflows/build.yml
+++ b/.github/workflows/build.yml
@@ -33,21 +33,24 @@ jobs:
           - "2.7.4.post1"
           - "2.8.3"
         python-version:
-          - "3.9"
-          # - "3.10"
-          # - "3.11"
-          # - "3.12"
+          # - "3.9"
+          - "3.10"
+          - "3.11"
+          - "3.12"
+          - "3.13"
         torch-version: 
-          - "2.5.1"
-          - "2.6.0"
-          - "2.7.1"
-          - "2.8.0"
+          # - "2.5.1"
+          # - "2.6.0"
+          # - "2.7.1"
+          # - "2.8.0"
+          - "2.9.0"
         # https://developer.nvidia.com/cuda-toolkit-archive
         cuda-version: 
-          - "12.4.1"
+          # - "12.4.1"
           - "12.6.3"
-          # - "12.8.1"
+          - "12.8.1"
           # - "12.9.1"
+          - "13.0.2"
         exclude:
           # torch < 2.2 does not support Python 3.12
           - python-version: "3.12"
@@ -78,6 +81,9 @@ jobs:
             cuda-version: "12.9.1"
           # flash-attn 2.7.4 does not build in GitHub Hosted Runner
           - flash-attn-version: "2.7.4"
+          # torch >= 2.9 does not support Python 3.9
+          - torch-version: "2.9.0"
+            python-version: "3.9"
     uses: ./.github/workflows/build_linux.yml
     with:
       flash-attn-version: ${{ matrix.flash-attn-version }}
diff --git a/get_torch_cuda_version.py b/get_torch_cuda_version.py
index c8a8adf..76c0124 100644
--- a/get_torch_cuda_version.py
+++ b/get_torch_cuda_version.py
@@ -10,7 +10,7 @@ support_cuda_versions = {
     "2.6": [118, 124, 126],
     "2.7": [118, 126, 128],
     "2.8": [126, 128, 129],
-    "2.9": [126, 128, 129],
+    "2.9": [126, 128, 130],
 }
 
 cuda_version = int(sys.argv[1])