ViperEkura
c52a2487ae
- all three release builds (pure, cu128, cu130) produce the same .whl filename, causing uploads to overwrite each other - append the CUDA tag as a local version label (e.g. +cu128, +cu130)
104 lines
2.7 KiB
YAML
104 lines
2.7 KiB
YAML
name: Release
|
|
|
|
on:
|
|
push:
|
|
tags:
|
|
- "v*"
|
|
|
|
jobs:
|
|
build-pure:
|
|
name: Build pure-Python wheel
|
|
runs-on: ubuntu-latest
|
|
steps:
|
|
- uses: actions/checkout@v4
|
|
- uses: actions/setup-python@v5
|
|
with:
|
|
python-version: "3.12"
|
|
|
|
- name: Build wheel (no CUDA)
|
|
run: |
|
|
pip wheel . --no-deps -w dist/
|
|
|
|
- uses: actions/upload-artifact@v4
|
|
with:
|
|
name: pure-wheel
|
|
path: dist/*.whl
|
|
if-no-files-found: error
|
|
|
|
build-cuda-linux:
|
|
name: Build CUDA wheel (Linux, ${{ matrix.cuda_tag }})
|
|
runs-on: ubuntu-latest
|
|
strategy:
|
|
fail-fast: false
|
|
matrix:
|
|
include:
|
|
- cuda_tag: "cu128"
|
|
cuda_ver: "12.8.0"
|
|
- cuda_tag: "cu130"
|
|
cuda_ver: "13.0.0"
|
|
steps:
|
|
- uses: actions/checkout@v4
|
|
- uses: actions/setup-python@v5
|
|
with:
|
|
python-version: "3.12"
|
|
|
|
- name: Install torch (${{ matrix.cuda_tag }})
|
|
run: |
|
|
pip install torch --index-url https://download.pytorch.org/whl/${{ matrix.cuda_tag }}
|
|
|
|
- name: Setup CUDA (${{ matrix.cuda_ver }})
|
|
uses: Jimver/cuda-toolkit@v0.2.35
|
|
with:
|
|
cuda: "${{ matrix.cuda_ver }}"
|
|
|
|
- name: Build wheel (with CUDA kernels)
|
|
run: |
|
|
CSRC_KERNELS=true pip wheel . --no-deps --no-build-isolation -w dist/
|
|
for f in dist/*.whl; do
|
|
mv "$f" "dist/$(basename "$f" .whl)+${{ matrix.cuda_tag }}.whl"
|
|
done
|
|
|
|
- uses: actions/upload-artifact@v4
|
|
with:
|
|
name: cuda-wheel-linux-${{ matrix.cuda_tag }}
|
|
path: dist/*.whl
|
|
if-no-files-found: error
|
|
|
|
release:
|
|
name: Attach wheels to release
|
|
needs: [build-pure, build-cuda-linux]
|
|
runs-on: ubuntu-latest
|
|
permissions:
|
|
contents: write
|
|
steps:
|
|
- name: Download pure-Python wheel
|
|
uses: actions/download-artifact@v4
|
|
with:
|
|
name: pure-wheel
|
|
path: release-assets/pure
|
|
|
|
- name: Download CUDA wheels (all variants)
|
|
uses: actions/download-artifact@v4
|
|
with:
|
|
pattern: cuda-wheel-linux-*
|
|
merge-multiple: true
|
|
path: release-assets/cuda
|
|
|
|
- name: Verify release assets
|
|
shell: bash
|
|
run: |
|
|
set -euo pipefail
|
|
pure_wheels=(release-assets/pure/*.whl)
|
|
cuda_wheels=(release-assets/cuda/*.whl)
|
|
test "${#pure_wheels[@]}" -eq 1
|
|
test "${#cuda_wheels[@]}" -ge 1
|
|
|
|
- name: Create release & upload assets
|
|
uses: softprops/action-gh-release@v2
|
|
with:
|
|
files: |
|
|
release-assets/pure/*.whl
|
|
release-assets/cuda/*.whl
|
|
tag_name: ${{ github.ref_name }}
|
|
generate_release_notes: true
|