Build Release and Publish #701
This file contains hidden or 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: 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 Build | |
| # Elevates permissions for `GITHUB_TOKEN`'s content scope (to allow this action to make a release) | |
| permissions: | |
| contents: write | |
| # Don't run this in everyone's forks on schedule but allow to run via dispatch. | |
| if: ${{ github.repository_owner == 'llvm' || github.event_name != 'schedule' }} | |
| strategy: | |
| matrix: | |
| package: [torch-mlir] | |
| arch: [x86_64, AArch64] | |
| py_version: ['3.10', '3.11', '3.12'] | |
| include: | |
| - py_version: '3.10' | |
| py_tag: 'cp310-cp310' | |
| - py_version: '3.11' | |
| py_tag: 'cp311-cp311' | |
| - py_version: '3.12' | |
| py_tag: 'cp312-cp312' | |
| runs-on: ${{ matrix.arch == 'AArch64' && 'ubuntu-24.04-arm' || 'ubuntu-24.04' }} | |
| steps: | |
| - name: Checkout torch-mlir | |
| uses: actions/checkout@8e8c483db84b4bee98b60c0593521ed34d9990e8 # v6.0.1 | |
| with: | |
| repository: llvm/torch-mlir | |
| ref: refs/heads/main | |
| submodules: 'true' | |
| - name: Setup workspace | |
| uses: ./.github/actions/setup-build | |
| with: | |
| python-version: ${{ matrix.py_version }} | |
| - name: Verify python version | |
| run: | | |
| # extract python major.minor version, for example, extract 3.10 from version 3.10.12 | |
| actual_version=$(python --version | cut -d' ' -f2 | cut -d'.' -f1,2) | |
| expected_version="${{ matrix.py_version }}" | |
| if [ "$actual_version" != "$expected_version" ]; then | |
| echo "ERROR: Python version mismatch! Expected $expected_version but got $actual_version" | |
| exit 1 | |
| fi | |
| echo "Python version verified: Expected $expected_version and got $actual_version" | |
| - 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_tag }} 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@b7c566a772e6b6bfb58ed0dc250532a479d7789f # v6.0.0 | |
| with: | |
| if-no-files-found: error | |
| name: snapshot-${{ matrix.package }}-${{ matrix.arch }}-${{ matrix.py_tag }}-${{ env.tm_package_version }} | |
| path: dist | |
| - name: Release python wheels | |
| uses: ncipollo/release-action@b7eabc95ff50cbeeedec83973935c8f306dfcd0b # v.1.20.0 | |
| with: | |
| artifacts: dist/*.whl | |
| token: "${{ secrets.GITHUB_TOKEN }}" | |
| tag: "dev-wheels" | |
| name: "dev-wheels" | |
| body: "Automatic snapshot release of torch-mlir python wheels." | |
| removeArtifacts: false | |
| allowUpdates: true | |
| replacesArtifacts: true | |
| makeLatest: true | |
| artifactErrorsFailBuild: true |