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