Skip to content

Commit 9116098

Browse files
committed
fix: matrix.config
1 parent abb0b50 commit 9116098

1 file changed

Lines changed: 7 additions & 16 deletions

File tree

.github/workflows/build.yml

Lines changed: 7 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -1,14 +1,12 @@
11
name: Wheel Build
2-
32
on:
43
workflow_dispatch:
54
push:
65
branches:
76
- ci
8-
97
jobs:
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

@@ -86,7 +78,6 @@ jobs:
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

Comments
 (0)