Minor buildsystem fixes (#778)

Sets up auto-pinning of latest torch-nightly
pull/780/head
powderluv 2022-04-21 15:53:00 -07:00 committed by GitHub
parent 0257d91a21
commit 4ef61aa27f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
6 changed files with 7 additions and 10 deletions

View File

@ -1,10 +1,6 @@
name: Manylinux Build
name: Release Build
on:
push:
branches:
- main
pull_request:
workflow_dispatch:
inputs:
release_id:

View File

@ -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 }}"}'

View File

@ -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

View File

@ -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}"

View File

@ -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"

View File

@ -1,8 +1,8 @@
-f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html
--pre
torch
numpy
torch
torchvision
# Build requirements.