ci: run

Author: souhhmm

.github/workflows/build.yml

@@ -2,14 +2,18 @@ name: Wheel Build
 
 on:
   workflow_dispatch:
+  push:
+    branches:
+      - ci
 
 jobs:
   build-wheel:
-    name: Build (PyTorch ${{ matrix.torch_version }}, CUDA ${{ matrix.cuda_tag }})
+    name: Build (torch${{ matrix.torch_version }}+${{ matrix.cuda_tag }}+py${{ matrix.python-version }})
     runs-on: ubuntu-22.04
     strategy:
       fail-fast: false
       matrix:
+        python-version: ["3.9", "3.10", "3.11", "3.12", "3.13", "3.14"]
         include:
           - torch_version: "2.10.0"
             cuda_version: "13.0.0"
@@ -39,10 +43,10 @@ jobs:
         with:
           submodules: recursive
 
-      - name: Set up Python 3.10
+      - name: Set up Python ${{ matrix.python-version }}
         uses: actions/setup-python@v5
         with:
-          python-version: "3.10"
+          python-version: "${{ matrix.python-version }}"
 
       - name: Install CUDA ${{ matrix.cuda_version }}
         uses: Jimver/cuda-toolkit@v0.2.30
@@ -61,7 +65,7 @@ jobs:
       - name: Build Wheel
         env:
           MAX_JOBS: 2
-          TORCH_CUDA_ARCH_LIST: "8.0 9.0"
+          TORCH_CUDA_ARCH_LIST: "7.0 7.5 8.0 8.6 8.9 9.0+PTX"
         run: |
           export PATH=/usr/local/cuda/bin:$PATH
           
@@ -73,7 +77,7 @@ jobs:
       - name: Upload Wheel Artifact
         uses: actions/upload-artifact@v4
         with:
-          name: lrnnx-wheel-torch${{ matrix.torch_version }}-${{ matrix.cuda_tag }}
+          name: lrnnx-wheel-torch${{ matrix.torch_version }}-${{ matrix.cuda_tag }}-py${{ matrix.python-version }}
           path: dist/*.whl
           retention-days: 7