Wheels #20
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: | | |
| sudo rm /usr/local/cuda | |
| 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_PATH=/usr/local/cuda | |
| export PATH="/usr/local/cuda/bin:$PATH" | |
| export LD_LIBRARY_PATH="/usr/local/cuda/lib:$LD_LIBRARY_PATH" | |
| export LD_LIBRARY_PATH="/usr/local/cuda/lib64:$LD_LIBRARY_PATH" | |
| export CUDA_HOME=${CUDA_PATH} | |
| echo "Adding CUDA to CUDA_PATH, PATH and LD_LIBRARY_PATH" | |
| echo "CUDA_PATH=/usr/local/cuda" >> $GITHUB_ENV | |
| echo "/usr/local/cuda/bin" >> $GITHUB_PATH | |
| echo "LD_LIBRARY_PATH=/usr/local/cuda/lib:${LD_LIBRARY_PATH}" >> $GITHUB_ENV | |
| echo "LD_LIBRARY_PATH=/usr/local/cuda/lib64:${LD_LIBRARY_PATH}" >> $GITHUB_ENV | |
| echo "CUDA_HOME=/usr/local/cuda" >> $GITHUB_ENV | |
| # echo "Adding CUDA to CUDA_PATH, PATH and LD_LIBRARY_PATH" | |
| # echo "CUDA_PATH=${CUDA_PATH}" >> $GITHUB_ENV | |
| # echo "${CUDA_PATH}/bin" >> $GITHUB_PATH | |
| # echo "LD_LIBRARY_PATH=${CUDA_PATH}/lib:${LD_LIBRARY_PATH}" >> $GITHUB_ENV | |
| # echo "LD_LIBRARY_PATH=${CUDA_PATH}/lib64:${LD_LIBRARY_PATH}" >> $GITHUB_ENV | |
| # 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 | |
| - name: PRINT | |
| run: | | |
| nvcc -V | |
| echo "CUDA_PATH=$CUDA_PATH" | |
| echo "CUDA_HOME=$CUDA_HOME" | |
| echo "LD_LIBRARY_PATH=$LD_LIBRARY_PATH" | |
| echo "PATH=$PATH" | |
| echo Readlink after $(readlink /usr/local/cuda) | |
| # echo "Adding CUDA to CUDA_PATH, PATH and LD_LIBRARY_PATH" | |
| # echo "CUDA_PATH=${CUDA_PATH}" >> $GITHUB_ENV | |
| # echo "${CUDA_PATH}/bin" >> $GITHUB_PATH | |
| # echo "LD_LIBRARY_PATH=${CUDA_PATH}/lib:${LD_LIBRARY_PATH}" >> $GITHUB_ENV | |
| # echo "LD_LIBRARY_PATH=${CUDA_PATH}/lib64:${LD_LIBRARY_PATH}" >> $GITHUB_ENV | |
| # echo "/usr/local/cuda/bin" >> $GITHUB_PATH | |
| nvcc -V | |
| - 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 }} |