Skip to content

Commit

Permalink
actions for publishing
Browse files Browse the repository at this point in the history
minor

fix py_version

fix

test hendrik/ccache, and split py3.10 and py3.11 insto separate jobs

include torch wheels too, add release body message

remove ccache entirely
  • Loading branch information
sjain-stanford committed Feb 8, 2024
1 parent fd84f68 commit 0e9f30d
Show file tree
Hide file tree
Showing 2 changed files with 86 additions and 0 deletions.
15 changes: 15 additions & 0 deletions .github/actions/setup-build/action.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,15 @@
name: "Setup build environment"
description: "Setup python, ninja-build etc."

runs:
using: "composite"

steps:
- name: Set up Python
uses: actions/setup-python@v5
with:
python-version: '3.11'

- name: Install prerequisites (Linux)
run: sudo apt-get install --yes ninja-build
shell: bash
71 changes: 71 additions & 0 deletions .github/workflows/buildReleaseAndPublish.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,71 @@
name: Build Release and Publish

on:
workflow_dispatch:
schedule:
# Runs at 11:00 AM UTC, which is 3:00 AM PST (UTC-8)
- cron: '0 11 * * *'

# Ensure that only a single job or workflow using the same
# concurrency group will run at a time. This would cancel
# any in-progress jobs in the same github workflow and github
# ref (e.g. refs/heads/main or refs/pull/<pr_number>/merge).
concurrency:
group: ${{ github.workflow }}-${{ github.ref }}
cancel-in-progress: true

jobs:
build_linux:
name: Linux x86_64 Build
runs-on: ubuntu-latest
# Don't run this in everyone's forks.
# TODO: FIXME
if: github.repository == 'sjain-stanford/torch-mlir-release'
strategy:
matrix:
package: [torch-mlir]
py_version: [cp310-cp310, cp311-cp311]

steps:
- name: Checkout torch-mlir
uses: actions/checkout@v4
with:
repository: llvm/torch-mlir
ref: refs/heads/main
submodules: 'true'

- name: Setup workspace
uses: ./.github/actions/setup-build

- name: Build Python wheels and smoke test
run: |
cd $GITHUB_WORKSPACE
tm_package_version="$(printf '%(%Y%m%d)T.${{ github.run_number }}')"
echo "tm_package_version=${tm_package_version}" >> $GITHUB_ENV
printf "TORCH_MLIR_PYTHON_PACKAGE_VERSION=%s\n" ${tm_package_version} > ./torch_mlir_package_version
TM_PYTHON_VERSIONS=${{ matrix.py_version }} TM_PACKAGES=${{ matrix.package }} ./build_tools/python_deploy/build_linux_packages.sh
- name: Make assets available in dist
run: |
mkdir dist
cp build_tools/python_deploy/wheelhouse/torch*.whl dist/
- name: Upload python wheels
uses: actions/upload-artifact@v4
with:
if-no-files-found: error
name: snapshot-${{ matrix.package }}-${{ matrix.py_version }}-${{ env.tm_package_version }}
path: dist

- name: Release python wheels
uses: ncipollo/release-action@v1.14.0
with:
artifacts: dist/*.whl
token: "${{ secrets.PUBLISH_ACCESS_TOKEN }}"
tag: "dev-wheels"
name: "dev-wheels"
body: "Automatic snapshot release of torch-mlir python wheels."
removeArtifacts: false
allowUpdates: true
replacesArtifacts: true
makeLatest: true

0 comments on commit 0e9f30d

Please sign in to comment.