chore: support torch2.9, bump the version to 1.0.2 (#781)

Author: lmxyy

.github/workflows/nightly-build.yaml

@@ -77,10 +77,7 @@ jobs:
     strategy:
       matrix:
         python: ["3.10", "3.11", "3.12", "3.13"]
-        torch: ["2.5", "2.6", "2.7", "2.8", "2.9"]
-        exclude:
-          - python: "3.13"
-            torch: "2.5"
+        torch: ["2.7", "2.8", "2.9", "pre"]
     steps:
       - name: Checkout to the tag
         uses: actions/checkout@v4
@@ -92,12 +89,12 @@ jobs:
         run: git log -1 --oneline
       - name: Build wheels
         run: |
-          if [[ "${{ matrix.torch }}" == "2.7" || "${{ matrix.torch }}" == "2.8" ]]; then
+          if [[ "${{ matrix.torch }}" == "2.7" || "${{ matrix.torch }}" == "2.8" || "${{ matrix.torch }}" == "2.9" ]]; then
             cuda_version="12.8"
           else
             cuda_version="12.4"
           fi
-          if [[ "${{ matrix.torch }}" == "2.9" ]]; then
+          if [[ "${{ matrix.torch }}" == "pre" ]]; then
             bash scripts/build_linux_wheel_torch_nightly.sh ${{ matrix.python }} ${{ matrix.torch }} 12.8
           else
             bash scripts/build_linux_wheel.sh ${{ matrix.python }} ${{ matrix.torch }} $cuda_version
@@ -122,10 +119,7 @@ jobs:
     strategy:
       matrix:
         python: ["3.10", "3.11", "3.12", "3.13"]
-        torch: ["2.5", "2.6", "2.7", "2.8", "2.9"]
-        exclude:
-          - python: "3.13"
-            torch: "2.5"
+        torch: ["2.7", "2.8", "2.9", "pre"]
     steps:
       - name: Checkout to the tag
         uses: actions/checkout@v4
@@ -144,11 +138,13 @@ jobs:
             SET CUDA_VERSION=12.8
           ) ELSE IF "%TORCH_VERSION%"=="2.8" (
             SET CUDA_VERSION=12.8
+          ) ELSE IF "%TORCH_VERSION%"=="2.9" (
+            SET CUDA_VERSION=12.8
           ) ELSE (
             SET CUDA_VERSION=12.4
           )
           call C:\Users\muyangl\miniconda3\condabin\activate.bat activate
-          IF "%TORCH_VERSION%"=="2.9" (
+          IF "%TORCH_VERSION%"=="pre" (
             call scripts\build_windows_wheel_torch_nightly.cmd %PYTHON_VERSION% %TORCH_VERSION% %CUDA_VERSION%
           ) ELSE (
             call scripts\build_windows_wheel.cmd %PYTHON_VERSION% %TORCH_VERSION% %CUDA_VERSION%

.github/workflows/release-build.yaml

@@ -51,10 +51,7 @@ jobs:
     strategy:
       matrix:
         python: ["3.10", "3.11", "3.12", "3.13"]
-        torch: ["2.5", "2.6", "2.7", "2.8", "2.9"]
-        exclude:
-          - python: "3.13"
-            torch: "2.5"
+        torch: ["2.7", "2.8", "2.9", "pre"]
     steps:
       - name: Checkout to the tag
         uses: actions/checkout@v4
@@ -66,12 +63,12 @@ jobs:
         run: git log -1 --oneline
       - name: Build wheels
         run: |
-          if [[ "${{ matrix.torch }}" == "2.7" || "${{ matrix.torch }}" == "2.8" ]]; then
+          if [[ "${{ matrix.torch }}" == "2.7" || "${{ matrix.torch }}" == "2.8" || "${{ matrix.torch }}" == "2.9" ]]; then
             cuda_version="12.8"
           else
             cuda_version="12.4"
           fi
-          if [[ "${{ matrix.torch }}" == "2.9" ]]; then
+          if [[ "${{ matrix.torch }}" == "pre" ]]; then
             bash scripts/build_linux_wheel_torch_nightly.sh ${{ matrix.python }} ${{ matrix.torch }} 12.8
           else
             bash scripts/build_linux_wheel.sh ${{ matrix.python }} ${{ matrix.torch }} $cuda_version
@@ -107,10 +104,7 @@ jobs:
     strategy:
       matrix:
         python: ["3.10", "3.11", "3.12", "3.13"]
-        torch: ["2.5", "2.6", "2.7", "2.8", "2.9"]
-        exclude:
-          - python: "3.13"
-            torch: "2.5"
+        torch: ["2.7", "2.8", "2.9", "pre"]
     steps:
       - name: Checkout to the tag
         uses: actions/checkout@v4
@@ -129,11 +123,13 @@ jobs:
             SET CUDA_VERSION=12.8
           ) ELSE IF "%TORCH_VERSION%"=="2.8" (
             SET CUDA_VERSION=12.8
+          ) ELSE IF "%TORCH_VERSION%"=="2.9" (
+            SET CUDA_VERSION=12.8
           ) ELSE (
             SET CUDA_VERSION=12.4
           )
           call C:\Users\muyangl\miniconda3\condabin\activate.bat activate
-          IF "%TORCH_VERSION%"=="2.9" (
+          IF "%TORCH_VERSION%"=="pre" (
             call scripts\build_windows_wheel_torch_nightly.cmd %PYTHON_VERSION% %TORCH_VERSION% %CUDA_VERSION%
           ) ELSE (
             call scripts\build_windows_wheel.cmd %PYTHON_VERSION% %TORCH_VERSION% %CUDA_VERSION%

nunchaku/__version__.py

@@ -1 +1 @@
-__version__ = "1.0.1"
+__version__ = "1.0.2"

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 requires = [
   "ninja",
   "setuptools",
-  "torch>=2.5",
+  "torch>=2.7",
   "wheel",
 ]
 

scripts/build_linux_wheel.sh

@@ -24,8 +24,12 @@ elif [ "$TORCH_VERSION" == "2.8" ]; then
   TORCHVISION_VERSION="0.23"
   TORCHAUDIO_VERSION="2.8"
   echo "TORCH_VERSION is 2.8, setting TORCHVISION_VERSION to $TORCHVISION_VERSION and TORCHAUDIO_VERSION to $TORCHAUDIO_VERSION"
+elif [ "$TORCH_VERSION" == "2.9" ]; then
+  TORCHVISION_VERSION="0.24"
+  TORCHAUDIO_VERSION="2.9"
+  echo "TORCH_VERSION is 2.9, setting TORCHVISION_VERSION to $TORCHVISION_VERSION and TORCHAUDIO_VERSION to $TORCHAUDIO_VERSION"
 else
-  echo "TORCH_VERSION is not 2.5, 2.6, 2.7 or 2.8, no changes to versions."
+  echo "TORCH_VERSION is not 2.5, 2.6, 2.7, 2.8 or 2.9, no changes to versions."
 fi
 
 docker run --rm \

scripts/build_windows_wheel.cmd

@@ -22,8 +22,11 @@ if "%TORCH_VERSION%"=="2.5" (
 ) else if "%TORCH_VERSION%"=="2.8" (
     set TORCHVISION_VERSION=0.23
     set TORCHAUDIO_VERSION=2.8
+) else if "%TORCH_VERSION%"=="2.9" (
+    set TORCHVISION_VERSION=0.24
+    set TORCHAUDIO_VERSION=2.9
 ) else (
-    echo TORCH_VERSION is not 2.5, 2.6, 2.7 or 2.8, no changes to versions.
+    echo TORCH_VERSION is not 2.5, 2.6, 2.7, 2.8 or 2.9, no changes to versions.
 )
 echo setting TORCHVISION_VERSION to %TORCHVISION_VERSION% and TORCHAUDIO_VERSION to %TORCHAUDIO_VERSION%