diff --git a/.github/scripts/validate_binaries.sh b/.github/scripts/validate_binaries.sh new file mode 100644 index 000000000..8984a6b59 --- /dev/null +++ b/.github/scripts/validate_binaries.sh @@ -0,0 +1,2 @@ +pip install ${PYTORCH_PIP_PREFIX} torchao --index-url ${PYTORCH_PIP_DOWNLOAD_URL} +python ./test/smoke_tests/smoke_tests.py diff --git a/.github/workflows/validate-binaries.yml b/.github/workflows/validate-binaries.yml new file mode 100644 index 000000000..087f3130f --- /dev/null +++ b/.github/workflows/validate-binaries.yml @@ -0,0 +1,47 @@ +name: Validate binaries + +on: + workflow_call: + inputs: + channel: + description: "Channel to use (nightly, test, release, all)" + required: false + type: string + default: release + ref: + description: "Reference to checkout, defaults to empty" + default: "" + required: false + type: string + workflow_dispatch: + inputs: + channel: + description: "Channel to use (nightly, test, release, all)" + required: true + type: choice + options: + - release + - nightly + - test + - all + ref: + description: "Reference to checkout, defaults to empty" + default: "" + required: false + type: string + pytorch_version: + description: "PyTorch version to validate (ie. 2.0, 2.2.2, etc.) - optional" + default: "" + required: false + type: string +jobs: + validate-binaries: + uses: pytorch/test-infra/.github/workflows/validate-domain-library.yml@main + with: + package_type: "wheel" + version: ${{ inputs.version }} + os: "linux" + channel: ${{ inputs.channel }} + repository: "pytorch/ao" + smoke_test: "source ./.github/scripts/validate_binaries.sh" + install_torch: true diff --git a/test/smoke_tests/smoke_tests.py b/test/smoke_tests/smoke_tests.py new file mode 100644 index 000000000..21537359f --- /dev/null +++ b/test/smoke_tests/smoke_tests.py @@ -0,0 +1,6 @@ +"""Run smoke tests""" + +import torchao + + +print("torchao version is ", torchao.__version__)