mirror of https://github.com/llvm/torch-mlir
parent
415265a64c
commit
ac1f03e6f7
|
@ -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.
|
||||
|
|
|
@ -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"
|
||||
|
|
Loading…
Reference in New Issue