add jit,no-jit release matrix (#1916)

pull/1918/head snapshot-20230306.769
Maksim Levental 2023-03-06 00:13:33 -06:00 committed by GitHub
parent 415265a64c
commit ac1f03e6f7
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 17 additions and 4 deletions

View File

@ -14,6 +14,16 @@ jobs:
build_linux:
name: Manylinux Build
runs-on: a100
strategy:
matrix:
package: [ torch-mlir, torch-mlir-no-jit ]
py_version: [ cp38-cp38, cp310-cp310, cp311-cp311 ]
exclude:
- package: torch-mlir-no-jit
py_version: cp38-cp38
- package: torch-mlir-no-jit
py_version: cp310-cp310
steps:
- name: Prepare workspace
@ -35,7 +45,7 @@ jobs:
python -m pip install wheel
TM_PACKAGE_VERSION=${{ github.event.inputs.python_package_version }}
printf "TORCH_MLIR_PYTHON_PACKAGE_VERSION=%s\n" $TM_PACKAGE_VERSION > ./torch_mlir_package_version
./build_tools/python_deploy/build_linux_packages.sh
TM_PYTHON_VERSIONS=${{ matrix.py_version }} TM_PACKAGES=${{ matrix.package }} ./build_tools/python_deploy/build_linux_packages.sh
# If we were given a release_id, then upload the package we just built
# to the github releases page.
@ -77,6 +87,9 @@ jobs:
build_macos:
name: MacOS Build
runs-on: macos-latest
strategy:
matrix:
package: [ torch-mlir, torch-mlir-no-jit ]
steps:
- name: Get torch-mlir
uses: actions/checkout@v3
@ -92,7 +105,7 @@ jobs:
TM_PACKAGE_VERSION=${{ github.event.inputs.python_package_version }}
printf "TORCH_MLIR_PYTHON_PACKAGE_VERSION=%s\n" $TM_PACKAGE_VERSION > ./torch_mlir_package_version
sudo ./build_tools/python_deploy/install_macos_deps.sh
TORCH_MLIR_PYTHON_VERSIONS="3.11" ./build_tools/python_deploy/build_macos_packages.sh
packages=${{ matrix.package }} TORCH_MLIR_PYTHON_VERSIONS="3.11" ./build_tools/python_deploy/build_macos_packages.sh
# If we were given a release_id, then upload the package we just built
# to the github releases page.

View File

@ -22,7 +22,7 @@ this_dir="$(cd "$(dirname "$0")" && pwd)"
repo_root="$(cd "$this_dir"/../../ && pwd)"
python_versions="${TORCH_MLIR_PYTHON_VERSIONS:-3.9 3.10 3.11}"
output_dir="${output_dir:-${this_dir}/wheelhouse}"
packages="${packages:-torch-mlir torch-mlir-no-jit-importer}"
packages="${packages:-torch-mlir}"
PKG_VER_FILE="${repo_root}"/torch_mlir_package_version ; [ -f "$PKG_VER_FILE" ] && . "$PKG_VER_FILE"
export TORCH_MLIR_PYTHON_PACKAGE_VERSION="${TORCH_MLIR_PYTHON_PACKAGE_VERSION:-0.0.1}"
@ -61,7 +61,7 @@ function run() {
build_torch_mlir torch_mlir "$python_version"
run_audit_wheel torch_mlir "$python_version"
;;
torch-mlir-no-jit-importer)
torch-mlir-no-jit)
clean_wheels torch_mlir_no_jit "$python_version"
build_torch_mlir_no_jit torch_mlir_no_jit "$python_version"
run_audit_wheel_no_jit torch_mlir_no_jit "$python_version"