mirror of https://github.com/llvm/torch-mlir
parent
0257d91a21
commit
4ef61aa27f
|
@ -1,10 +1,6 @@
|
|||
name: Manylinux Build
|
||||
name: Release Build
|
||||
|
||||
on:
|
||||
push:
|
||||
branches:
|
||||
- main
|
||||
pull_request:
|
||||
workflow_dispatch:
|
||||
inputs:
|
||||
release_id:
|
|
@ -55,10 +55,10 @@ jobs:
|
|||
ref: "${{ env.tag_name }}"
|
||||
inputs: '{"release_id": "", "python_package_version": "${{ env.package_version }}"}'
|
||||
|
||||
- name: "Invoke workflow :: Manylinux Build"
|
||||
- name: "Invoke workflow :: Release Build"
|
||||
uses: benc-uk/workflow-dispatch@v1
|
||||
with:
|
||||
workflow: Manylinux Build
|
||||
workflow: Release Build
|
||||
token: ${{ secrets.WORKFLOW_INVOCATION_TOKEN }}
|
||||
ref: "${{ env.tag_name }}"
|
||||
inputs: '{"release_id": "${{ steps.create_release.outputs.id }}", "python_package_version": "${{ env.package_version }}"}'
|
||||
|
|
|
@ -95,6 +95,7 @@ function run_in_docker() {
|
|||
}
|
||||
|
||||
function build_torch_mlir() {
|
||||
python -m pip install -r /main_checkout/torch-mlir/requirements.txt --extra-index-url https://download.pytorch.org/whl/nightly/cpu
|
||||
CMAKE_GENERATOR=Ninja \
|
||||
python -m pip wheel -v -w /wheelhouse /main_checkout/torch-mlir/ \
|
||||
--extra-index-url https://download.pytorch.org/whl/nightly/cpu
|
||||
|
|
|
@ -20,7 +20,7 @@ set -eu -o errtrace
|
|||
|
||||
this_dir="$(cd $(dirname $0) && pwd)"
|
||||
repo_root="$(cd $this_dir/../../ && pwd)"
|
||||
python_versions="${python_versions:3.9 3.10}"
|
||||
python_versions="${python_versions:-3.9 3.10}"
|
||||
output_dir="${output_dir:-${this_dir}/wheelhouse}"
|
||||
packages="${packages:-torch-mlir}"
|
||||
|
||||
|
|
|
@ -16,6 +16,6 @@ requires = [
|
|||
# artifacts.
|
||||
# TODO: Come up with a better way to pin the version.
|
||||
"numpy",
|
||||
"torch==1.12.0.dev20220420",
|
||||
"torch>=1.12.0.dev20220420",
|
||||
]
|
||||
build-backend = "setuptools.build_meta"
|
||||
|
|
|
@ -1,8 +1,8 @@
|
|||
-f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html
|
||||
--pre
|
||||
torch
|
||||
|
||||
numpy
|
||||
torch
|
||||
torchvision
|
||||
|
||||
# Build requirements.
|
||||
|
|
Loading…
Reference in New Issue