From e9f16ebc017ba8aefeada8edf720dc96086bb047 Mon Sep 17 00:00:00 2001
From: Junya Morioka <mjun@mjunya.com>
Date: Wed, 28 Jan 2026 18:17:12 +0900
Subject: [PATCH] feat: add missing wheel build matrices for Linux x86_64 and
 ARM64

- Expand LINUX_SELF_HOSTED_MATRIX with torch 2.6.0-2.9.1 and CUDA
  12.4/12.8/13.0 to cover 29 missing Linux x86_64 packages
- Add torch 2.9.1 to LINUX_ARM64_MATRIX for 9 missing ARM64 packages
- Enable linux_arm64 matrix in main() output
---
 create_matrix.py | 21 ++++++++++-----------
 1 file changed, 10 insertions(+), 11 deletions(-)

diff --git a/create_matrix.py b/create_matrix.py
index 446c225..bca4ea0 100644
--- a/create_matrix.py
+++ b/create_matrix.py
@@ -91,7 +91,7 @@ LINUX_ARM64_MATRIX = {
         # "2.6.0",
         # "2.7.1",
         # "2.8.0",
-        # "2.9.1",
+        "2.9.1",
         "2.10.0",
     ],
     "cuda-version": [
@@ -118,19 +118,18 @@ LINUX_SELF_HOSTED_MATRIX = {
     ],
     "torch-version": [
         # "2.5.1",
-        # "2.6.0",
-        # "2.7.1",
-        # "2.8.0",
-        # "2.9.1",
+        "2.6.0",
+        "2.7.1",
+        "2.8.0",
+        "2.9.1",
         "2.10.0",
     ],
     "cuda-version": [
-        # "12.4",
+        "12.4",
         "12.6",
-        # "12.8",
+        "12.8",
         # "12.9",
-        # "13.0",
-        # "13.1",
+        "13.0",
     ],
 }
 
@@ -254,8 +253,8 @@ def main():
                 "linux": False,
                 # "linux": LINUX_MATRIX,
                 #
-                "linux_arm64": False,
-                # "linux_arm64": LINUX_ARM64_MATRIX,
+                # "linux_arm64": False,
+                "linux_arm64": LINUX_ARM64_MATRIX,
                 #
                 # "linux_self_hosted": False,
                 "linux_self_hosted": LINUX_SELF_HOSTED_MATRIX,