111 lines
3.9 KiB
YAML
111 lines
3.9 KiB
YAML
# This workflow will upload a Python Package to Release asset
|
|
# For more information see: https://help.github.com/en/actions/language-and-framework-guides/using-python-with-github-actions
|
|
|
|
name: Create Release
|
|
|
|
on:
|
|
push:
|
|
tags:
|
|
- v*
|
|
|
|
# Needed to create release and upload assets
|
|
permissions:
|
|
contents: write
|
|
|
|
jobs:
|
|
release:
|
|
# Retrieve tag and create release
|
|
name: Create Release
|
|
runs-on: ubuntu-latest
|
|
outputs:
|
|
upload_url: ${{ steps.create_release.outputs.upload_url }}
|
|
steps:
|
|
- name: Checkout
|
|
uses: actions/checkout@eef61447b9ff4aafe5dcd4e0bbf5d482be7e7871 # v4.2.1
|
|
|
|
- name: Extract branch info
|
|
shell: bash
|
|
run: |
|
|
echo "release_tag=${GITHUB_REF#refs/*/}" >> "$GITHUB_ENV"
|
|
|
|
- name: Create Release
|
|
id: create_release
|
|
uses: actions/github-script@60a0d83039c74a4aee543508d2ffcb1c3799cdea # v7.0.1
|
|
env:
|
|
RELEASE_TAG: ${{ env.release_tag }}
|
|
with:
|
|
github-token: "${{ secrets.GITHUB_TOKEN }}"
|
|
script: |
|
|
const script = require('.github/workflows/scripts/create_release.js')
|
|
await script(github, context, core)
|
|
|
|
wheel:
|
|
name: Build Wheel
|
|
runs-on: ${{ matrix.os }}
|
|
needs: release
|
|
|
|
strategy:
|
|
fail-fast: false
|
|
matrix:
|
|
os: ['ubuntu-20.04']
|
|
python-version: ['3.9', '3.10', '3.11', '3.12']
|
|
pytorch-version: ['2.4.0'] # Must be the most recent version that meets requirements-cuda.txt.
|
|
cuda-version: ['11.8', '12.1']
|
|
|
|
steps:
|
|
- name: Checkout
|
|
uses: actions/checkout@eef61447b9ff4aafe5dcd4e0bbf5d482be7e7871 # v4.2.1
|
|
|
|
- name: Setup ccache
|
|
uses: hendrikmuhs/ccache-action@ed74d11c0b343532753ecead8a951bb09bb34bc9 # v1.2.14
|
|
with:
|
|
create-symlink: true
|
|
key: ${{ github.job }}-${{ matrix.python-version }}-${{ matrix.cuda-version }}
|
|
|
|
- name: Set up Linux Env
|
|
if: ${{ runner.os == 'Linux' }}
|
|
run: |
|
|
bash -x .github/workflows/scripts/env.sh
|
|
|
|
- name: Set up Python
|
|
uses: actions/setup-python@f677139bbe7f9c59b41e40162b753c062f5d49a3 # v5.2.0
|
|
with:
|
|
python-version: ${{ matrix.python-version }}
|
|
|
|
- name: Install CUDA ${{ matrix.cuda-version }}
|
|
run: |
|
|
bash -x .github/workflows/scripts/cuda-install.sh ${{ matrix.cuda-version }} ${{ matrix.os }}
|
|
|
|
- name: Install PyTorch ${{ matrix.pytorch-version }} with CUDA ${{ matrix.cuda-version }}
|
|
run: |
|
|
bash -x .github/workflows/scripts/pytorch-install.sh ${{ matrix.python-version }} ${{ matrix.pytorch-version }} ${{ matrix.cuda-version }}
|
|
|
|
- name: Build wheel
|
|
shell: bash
|
|
env:
|
|
CMAKE_BUILD_TYPE: Release # do not compile with debug symbol to reduce wheel size
|
|
run: |
|
|
bash -x .github/workflows/scripts/build.sh ${{ matrix.python-version }} ${{ matrix.cuda-version }}
|
|
wheel_name=$(find dist -name "*whl" -print0 | xargs -0 -n 1 basename)
|
|
asset_name=${wheel_name//"linux"/"manylinux1"}
|
|
echo "wheel_name=${wheel_name}" >> "$GITHUB_ENV"
|
|
echo "asset_name=${asset_name}" >> "$GITHUB_ENV"
|
|
|
|
- name: Upload Release Asset
|
|
uses: actions/upload-release-asset@e8f9f06c4b078e705bd2ea027f0926603fc9b4d5 # v1.0.2
|
|
env:
|
|
GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
|
|
with:
|
|
upload_url: ${{ needs.release.outputs.upload_url }}
|
|
asset_path: ./dist/${{ env.wheel_name }}
|
|
asset_name: ${{ env.asset_name }}
|
|
asset_content_type: application/*
|
|
|
|
# (Danielkinz): This last step will publish the .whl to pypi. Warning: untested
|
|
# - name: Publish package
|
|
# uses: pypa/gh-action-pypi-publish@release/v1.8
|
|
# with:
|
|
# repository-url: https://test.pypi.org/legacy/
|
|
# password: ${{ secrets.PYPI_API_TOKEN }}
|
|
# skip-existing: true
|