72 lines
1.6 KiB
YAML
72 lines
1.6 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
|
|
|
|
build-cuda-linux:
|
|
name: Build CUDA wheel (Linux)
|
|
runs-on: ubuntu-latest
|
|
steps:
|
|
- uses: actions/checkout@v4
|
|
- uses: actions/setup-python@v5
|
|
with:
|
|
python-version: "3.12"
|
|
|
|
- name: Install torch (CUDA 12.8)
|
|
run: |
|
|
pip install torch --index-url https://download.pytorch.org/whl/cu128
|
|
|
|
- name: Setup CUDA
|
|
uses: Jimver/cuda-toolkit@v0.2.35
|
|
with:
|
|
cuda: "12.8.0"
|
|
|
|
- name: Build wheel (with CUDA kernels)
|
|
run: |
|
|
CSRC_KERNELS=true pip wheel . --no-deps --no-build-isolation -w dist/
|
|
|
|
- uses: actions/upload-artifact@v4
|
|
with:
|
|
name: cuda-wheel-linux
|
|
path: dist/*.whl
|
|
|
|
release:
|
|
name: Attach wheels to release
|
|
needs: [build-pure, build-cuda-linux]
|
|
runs-on: ubuntu-latest
|
|
permissions:
|
|
contents: write
|
|
steps:
|
|
- uses: actions/download-artifact@v4
|
|
with:
|
|
pattern: "*-wheel"
|
|
merge-multiple: true
|
|
|
|
- name: Create release & upload assets
|
|
uses: softprops/action-gh-release@v2
|
|
with:
|
|
files: ./*.whl
|
|
tag_name: ${{ github.ref_name }}
|
|
generate_release_notes: true
|