From 15c6a2da4910cb0407253e560c0ad788a2fc414a Mon Sep 17 00:00:00 2001
From: LoserCheems <3314685395@qq.com>
Date: Tue, 9 Sep 2025 10:31:59 +0800
Subject: [PATCH] Update PyTorch version matrix to remove 2.4.0 and add support
 for 2.5.1 on Python 3.13

---
 .github/workflows/publish.yml | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/.github/workflows/publish.yml b/.github/workflows/publish.yml
index cd89e19..2adadf1 100644
--- a/.github/workflows/publish.yml
+++ b/.github/workflows/publish.yml
@@ -45,7 +45,7 @@ jobs:
         # manylinux docker image, but I haven't figured out how to install CUDA on manylinux.
         os: [ubuntu-22.04, ubuntu-22.04-arm]
         python-version: ["3.9", "3.10", "3.11", "3.12", "3.13"]
-        torch-version: ["2.4.0", "2.5.1", "2.6.0", "2.7.1", "2.8.0"]
+        torch-version: ["2.5.1", "2.6.0", "2.7.1", "2.8.0"]
         cuda-version: ["12.9.1"]
         # We need separate wheels that either uses C++11 ABI (-D_GLIBCXX_USE_CXX11_ABI) or not.
         # Pytorch wheels currently don't use it, but nvcr images have Pytorch compiled with C++11 ABI.
@@ -60,6 +60,9 @@ jobs:
           # Pytorch < 2.5 does not support Python 3.13
           - torch-version: "2.4.0"
             python-version: "3.13"
+          - torch-version: "2.5.1"
+            python-version: "3.13"
+            os: ubuntu-22.04-arm
     uses: ./.github/workflows/_build.yml
     with:
       runs-on: ${{ matrix.os }}