-
Notifications
You must be signed in to change notification settings - Fork 5
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Add support for releasing windows wheels. (#2)
This PR adds support for windows nightly build for releasing `torch-mlir` python wheels following the work done for linux. Here's the job for the windows build: https://github.com/sahas3/torch-mlir-release/actions/runs/11425775316
- Loading branch information
Showing
2 changed files
with
82 additions
and
2 deletions.
There are no files selected for viewing
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
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,80 @@ | ||
name: Build Release and Publish windows wheels | ||
|
||
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_x64_msvc: | ||
name: windows x86_64 msvc Build | ||
runs-on: windows-2022 | ||
defaults: | ||
run: | ||
shell: bash | ||
# 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. | ||
if: github.repository == 'llvm/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: "Configuring MSVC" | ||
uses: ilammy/[email protected] | ||
|
||
- name: Build torch-mlir | ||
run: | | ||
cd $GITHUB_WORKSPACE | ||
./build_tools/python_deploy/build_windows_ci.sh | ||
- name: Build Python wheels and smoke test | ||
run: | | ||
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 | ||
pip install delvewheel | ||
TORCH_MLIR_PYTHON_PACKAGE_VERSION=${tm_package_version} TORCH_MLIR_CMAKE_BUILD_DIR="build" TORCH_MLIR_CMAKE_ALREADY_BUILT=1 python setup.py bdist_wheel | ||
delvewheel repair --add-path ./build/tools/torch-mlir/python_packages/torch_mlir/torch_mlir/_mlir_libs --add-dll TorchMLIRAggregateCAPI.dll --no-dll 'c10.dll;torch_python.dll;torch_cpu.dll' -v dist/torch_mlir*.whl -w 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/[email protected] | ||
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 |
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