From c55e7183632e8005c0509f1c56ae98ac441a4305 Mon Sep 17 00:00:00 2001
From: Mohammad Miadh Angkad <MAngkad.BSDSBA2027@aim.edu>
Date: Thu, 15 Jan 2026 19:18:58 +0800
Subject: [PATCH] Add PyTorch 2.10 to all matrices and add cu129 for torch 2.9

---
 create_matrix.py          | 10 ++++++++++
 get_torch_cuda_version.py |  3 ++-
 2 files changed, 12 insertions(+), 1 deletion(-)

diff --git a/create_matrix.py b/create_matrix.py
index 647b281..2255600 100644
--- a/create_matrix.py
+++ b/create_matrix.py
@@ -21,12 +21,15 @@ EXCLUDE = [
     {"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-version": "2.10.0", "python-version": "3.9"},
     # torch < 2.9 does not support CUDA 13.0
     {"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"},
+    # torch 2.10 does not support CUDA 12.4
+    {"torch-version": "2.10.0", "cuda-version": "12.4"},
     # Python 3.14 is supported from torch 2.9
     {"torch-version": "2.5.1", "python-version": "3.14"},
     {"torch-version": "2.6.3", "python-version": "3.14"},
@@ -52,6 +55,7 @@ LINUX_MATRIX = {
         "2.7.1",
         "2.8.0",
         "2.9.1",
+        "2.10.0",
     ],
     "cuda-version": [
         "12.4",
@@ -80,6 +84,7 @@ LINUX_ARM64_MATRIX = {
         "2.7.1",
         # "2.8.0",
         "2.9.1",
+        "2.10.0",
     ],
     "cuda-version": [
         "12.4",
@@ -109,6 +114,7 @@ LINUX_SELF_HOSTED_MATRIX = {
         "2.7.1",
         "2.8.0",
         # "2.9.1",
+        "2.10.0",
     ],
     "cuda-version": [
         "12.4",
@@ -138,6 +144,7 @@ LINUX_ARM64_SELF_HOSTED_MATRIX = {
         # "2.7.1",
         # "2.8.0",
         "2.9.1",
+        "2.10.0",
     ],
     "cuda-version": [
         # "12.4",
@@ -166,6 +173,7 @@ WINDOWS_MATRIX = {
         # "2.7.1",
         # "2.8.0",
         "2.9.1",
+        "2.10.0",
     ],
     "cuda-version": [
         # "12.4",
@@ -190,6 +198,7 @@ WINDOWS_CODEBUILD_MATRIX = {
     ],
     "torch-version": [
         "2.9.1",
+        "2.10.0",
     ],
     "cuda-version": [
         "12.8",
@@ -215,6 +224,7 @@ WINDOWS_SELF_HOSTED_MATRIX = {
         "2.7.1",
         "2.8.0",
         "2.9.1",
+        "2.10.0",
     ],
     "cuda-version": [
         # "12.4",
diff --git a/get_torch_cuda_version.py b/get_torch_cuda_version.py
index 76c0124..faef11d 100644
--- a/get_torch_cuda_version.py
+++ b/get_torch_cuda_version.py
@@ -10,7 +10,8 @@ support_cuda_versions = {
     "2.6": [118, 124, 126],
     "2.7": [118, 126, 128],
     "2.8": [126, 128, 129],
-    "2.9": [126, 128, 130],
+    "2.9": [126, 128, 129, 130],
+    "2.10": [126, 128, 129, 130],
 }
 
 cuda_version = int(sys.argv[1])