From e6542f9d80edb7078f49c3d7c95cbf9617c5e1da Mon Sep 17 00:00:00 2001
From: Junya Morioka <mjun@mjunya.com>
Date: Fri, 5 Dec 2025 14:53:34 +0900
Subject: [PATCH] feat: Update arm matrix

---
 create_matrix.py | 100 +++++++++++++++++++++++++++++------------------
 1 file changed, 63 insertions(+), 37 deletions(-)

diff --git a/create_matrix.py b/create_matrix.py
index 421c645..f41c23e 100644
--- a/create_matrix.py
+++ b/create_matrix.py
@@ -5,26 +5,26 @@ EXCLUDE = [
     {"python-version": "3.12", "torch-version": "2.0.1"},
     {"python-version": "3.12", "torch-version": "2.1.2"},
     # torch 2.0.1 does not support CUDA 12.x
-    {"torch-version": "2.0.1", "cuda-version": "12.1.1"},
-    {"torch-version": "2.0.1", "cuda-version": "12.4.1"},
-    {"torch-version": "2.0.1", "cuda-version": "12.6.3"},
-    {"torch-version": "2.0.1", "cuda-version": "12.8.1"},
+    {"torch-version": "2.0.1", "cuda-version": "12.1"},
+    {"torch-version": "2.0.1", "cuda-version": "12.4"},
+    {"torch-version": "2.0.1", "cuda-version": "12.6"},
+    {"torch-version": "2.0.1", "cuda-version": "12.8"},
     # torch 2.6.0 does not support CUDA 12.1
-    {"torch-version": "2.6.0", "cuda-version": "12.1.1"},
+    {"torch-version": "2.6.0", "cuda-version": "12.1"},
     # torch 2.7.0 does not support CUDA 12.4
-    {"torch-version": "2.7.0", "cuda-version": "12.4.1"},
+    {"torch-version": "2.7.0", "cuda-version": "12.4"},
     # torch < 2.8 does not support CUDA 12.9
-    {"torch-version": "2.5.1", "cuda-version": "12.9.1"},
-    {"torch-version": "2.6.3", "cuda-version": "12.9.1"},
-    {"torch-version": "2.7.1", "cuda-version": "12.9.1"},
+    {"torch-version": "2.5.1", "cuda-version": "12.9"},
+    {"torch-version": "2.6.3", "cuda-version": "12.9"},
+    {"torch-version": "2.7.1", "cuda-version": "12.9"},
     # torch >= 2.9 does not support Python 3.9
     {"torch-version": "2.9.1", "python-version": "3.9"},
     # torch < 2.9 does not support CUDA 13.0
-    {"torch-version": "2.5.1", "cuda-version": "13.0.1"},
-    {"torch-version": "2.6.0", "cuda-version": "13.0.1"},
-    {"torch-version": "2.7.1", "cuda-version": "13.0.1"},
-    {"torch-version": "2.8.1", "cuda-version": "13.0.1"},
-    {"torch-version": "2.8.0", "cuda-version": "13.0.1"},
+    {"torch-version": "2.5.1", "cuda-version": "13.0"},
+    {"torch-version": "2.6.0", "cuda-version": "13.0"},
+    {"torch-version": "2.7.1", "cuda-version": "13.0"},
+    {"torch-version": "2.8.1", "cuda-version": "13.0"},
+    {"torch-version": "2.8.0", "cuda-version": "13.0"},
 ]
 
 LINUX_MATRIX = {
@@ -47,28 +47,19 @@ LINUX_MATRIX = {
         "2.9.1",
     ],
     "cuda-version": [
-        "12.4.1",
-        # "12.6.3",
-        "12.8.1",
-        # "12.9.1",
-        "13.0.2",
+        "12.4",
+        # "12.6",
+        "12.8",
+        # "12.9",
+        "13.0",
     ],
 }
 
-LINUX_ARM64_MATRIX = LINUX_MATRIX
-
-LINUX_SELF_HOSTED_MATRIX = {
-    "flash-attn-version": ["2.7.4"],
-    "python-version": ["3.10", "3.11", "3.12", "3.13"],
-    "torch-version": ["2.9.1"],
-    "cuda-version": ["12.8.1", "13.0.1"],
-}
-
-WINDOWS_MATRIX = {
+LINUX_ARM64_MATRIX = {
     "flash-attn-version": [
         "2.6.3",
-        "2.7.4",
-        "2.8.3",
+        # "2.7.4",
+        # "2.8.3",
     ],
     "python-version": [
         "3.10",
@@ -84,11 +75,46 @@ WINDOWS_MATRIX = {
         "2.9.1",
     ],
     "cuda-version": [
-        "12.4.1",
-        "12.6.3",
-        "12.8.1",
-        # "12.9.1",
-        "13.0.2",
+        "12.4",
+        # "12.6",
+        "12.8",
+        # "12.9",
+        "13.0",
+    ],
+}
+
+LINUX_SELF_HOSTED_MATRIX = {
+    "flash-attn-version": ["2.7.4"],
+    "python-version": ["3.10", "3.11", "3.12", "3.13"],
+    "torch-version": ["2.9.1"],
+    "cuda-version": ["12.8", "13.0"],
+}
+
+WINDOWS_MATRIX = {
+    "flash-attn-version": [
+        "2.6.3",
+        # "2.7.4",
+        # "2.8.3",
+    ],
+    "python-version": [
+        "3.10",
+        "3.11",
+        "3.12",
+        # "3.13",
+    ],
+    "torch-version": [
+        "2.5.1",
+        "2.6.0",
+        "2.7.1",
+        "2.8.0",
+        "2.9.1",
+    ],
+    "cuda-version": [
+        "12.4",
+        # "12.6",
+        "12.8",
+        # "12.9",
+        "13.0",
     ],
 }
 
@@ -96,7 +122,7 @@ WINDOWS_CODEBUILD_MATRIX = {
     "flash-attn-version": ["2.6.3", "2.7.4.post1", "2.8.3"],
     "python-version": ["3.10", "3.11", "3.12", "3.13"],
     "torch-version": ["2.9.1"],
-    "cuda-version": ["13.0.1"],
+    "cuda-version": ["13.0"],
 }