Wheels #15
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
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 }} |