11name : Wheel Build
2-
32on :
43 workflow_dispatch :
54 push :
65 branches :
76 - ci
8-
97jobs :
108 build-wheel :
11- name : Build (torch${{ matrix.torch_version }}+${{ matrix.cuda_tag }}+py${{ matrix.python-version }})
9+ name : Build (torch${{ matrix.config. torch_version }}+${{ matrix.config .cuda_tag }}+py${{ matrix.python-version }})
1210 runs-on : ubuntu-22.04
1311 strategy :
1412 fail-fast : false
@@ -18,25 +16,21 @@ jobs:
1816 - torch_version : " 2.10.0"
1917 cuda_version : " 13.0.0"
2018 cuda_tag : " cu130"
21-
2219 - torch_version : " 2.9.1"
2320 cuda_version : " 13.0.0"
2421 cuda_tag : " cu130"
2522 - torch_version : " 2.9.1"
2623 cuda_version : " 12.8.1"
2724 cuda_tag : " cu128"
28-
2925 - torch_version : " 2.8.0"
3026 cuda_version : " 12.9.1"
3127 cuda_tag : " cu129"
3228 - torch_version : " 2.8.0"
3329 cuda_version : " 12.8.1"
3430 cuda_tag : " cu128"
35-
3631 - torch_version : " 2.7.1"
37- cuda_version : " 12.8.1"
32+ cuda_version : " 12.8.1"
3833 cuda_tag : " cu128"
39-
4034 steps :
4135 - name : Checkout repository
4236 uses : actions/checkout@v4
@@ -48,36 +42,34 @@ jobs:
4842 with :
4943 python-version : " ${{ matrix.python-version }}"
5044
51- - name : Install CUDA ${{ matrix.cuda_version }}
45+ - name : Install CUDA ${{ matrix.config. cuda_version }}
5246 uses : Jimver/cuda-toolkit@v0.2.30
5347 id : cuda-toolkit
5448 with :
55- cuda : " ${{ matrix.cuda_version }}"
49+ cuda : " ${{ matrix.config. cuda_version }}"
5650 linux-local-args : ' ["--toolkit"]'
5751 method : " network"
5852
5953 - name : Install PyTorch & Build Dependencies
6054 run : |
6155 pip install --upgrade pip
6256 pip install setuptools==68.1.0 ninja packaging wheel
63- pip install torch==${{ matrix.torch_version }} --index-url https://download.pytorch.org/whl/${{ matrix.cuda_tag }}
57+ pip install torch==${{ matrix.config. torch_version }} --index-url https://download.pytorch.org/whl/${{ matrix.config .cuda_tag }}
6458
6559 - name : Build Wheel
6660 env :
6761 MAX_JOBS : 2
6862 TORCH_CUDA_ARCH_LIST : " 7.0 7.5 8.0 8.6 8.9 9.0+PTX"
6963 run : |
7064 export PATH=/usr/local/cuda/bin:$PATH
71-
72- LOCAL_TAG="+${{ matrix.cuda_tag }}torch${{ matrix.torch_version }}"
73-
65+ LOCAL_TAG="+${{ matrix.config.cuda_tag }}torch${{ matrix.config.torch_version }}"
7466 timeout 5h python setup.py egg_info --tag-build="$LOCAL_TAG" bdist_wheel --dist-dir=dist
7567 ls -lh dist/
7668
7769 - name : Upload Wheel Artifact
7870 uses : actions/upload-artifact@v4
7971 with :
80- name : lrnnx-wheel-torch${{ matrix.torch_version }}-${{ matrix.cuda_tag }}-py${{ matrix.python-version }}
72+ name : lrnnx-wheel-torch${{ matrix.config. torch_version }}-${{ matrix.config .cuda_tag }}-py${{ matrix.python-version }}
8173 path : dist/*.whl
8274 retention-days : 7
8375
8678 runs-on : ubuntu-22.04
8779 needs : build-wheel
8880 if : always()
89-
9081 steps :
9182 - name : Checkout repository
9283 uses : actions/checkout@v4
0 commit comments