Skip to content

Wheels

Wheels #15

Workflow file for this run

name: Wheels
on:
workflow_dispatch:
pull_request:
# push:
# branches:
# - main
release:
types:
- published
jobs:
build_sdist:
name: Build SDist
runs-on: ubuntu-latest
steps:
- uses: actions/checkout@v4
with:
submodules: true
- name: Build SDist
run: pipx run build --sdist
- name: Check metadata
run: pipx run twine check dist/*
- uses: actions/upload-artifact@v4
with:
name: dist-sdist
path: dist/*.tar.gz
build_wheels:
name: Wheels on ${{ matrix.os }}
runs-on: ${{ matrix.os }}
strategy:
fail-fast: false
matrix:
os: [ubuntu-latest]
steps:
- name: Fetch CUDA toolkit
uses: Jimver/[email protected]
id: cuda-toolkit
with:
linux-local-args: '["--toolkit"]'
method: 'network'
sub-packages: '["nvcc"]'
- name: Install CUDA (Linux only)
if: ${{ runner.os == 'Linux' }}
run: |
echo "CUDA PATH: $CUDA_HOME"
echo "LD LIBRARY $LD_LIBRARY_PATH"
echo "PATH $PATH"
echo Readlink before $(readlink /usr/local/cuda)
sudo ln -s ${{steps.cuda-toolkit.outputs.CUDA_PATH}} /usr/local/cuda
echo Readlink after $(readlink /usr/local/cuda)
export CUDA_HOME=/usr/local/cuda
export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/bin:$LD_LIBRARY_PATH
export CUDA_BIN_PATH=${{steps.cuda-toolkit.outputs.CUDA_PATH}}
echo "/usr/local/cuda/bin" >> $GITHUB_PATH
nvcc -V
# - name: Install CUDA (Windows only)
# if: ${{ runner.os == 'Windows' }}
# run: |
# $CUDA_PATH=Get-ChildItem "C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA" -filter "v*.*" -Directory | % { $_.fullname }
# echo "CUDA_PATH=$CUDA_PATH" >> $env:GITHUB_ENV
# echo "CUDA_TOOLKIT_ROOT_DIR=$CUDA_PATH" >> $env:GITHUB_ENV
- uses: actions/checkout@v4
with:
submodules: true
- uses: pypa/[email protected]
- name: Verify clean directory
run: git diff --exit-code
shell: bash
- name: Upload wheels
uses: actions/upload-artifact@v4
with:
path: wheelhouse/*.whl
name: dist-${{ matrix.os }}
merge_wheels:
name: Merge wheels into a combined artifact
runs-on: ubuntu-latest
needs: [build_wheels, build_sdist]
steps:
- name: Merge Artifacts
uses: actions/upload-artifact/merge@v4
with:
name: dist
pattern: dist-*
upload_all:
name: Upload if release
needs: merge_wheels
runs-on: ubuntu-latest
if: github.event_name == 'release' && github.event.action == 'published'
steps:
- uses: actions/setup-python@v5
- uses: actions/download-artifact@v4
with:
path: dist
- uses: pypa/gh-action-pypi-publish@release/v1
with:
user: __token__
password: ${{ secrets.pypi_password }}