From b9889d6cad10a6612cf7fce34b03bf8293005a55 Mon Sep 17 00:00:00 2001 From: atalman Date: Wed, 6 Dec 2023 14:59:44 -0800 Subject: [PATCH] Validation scripts, install using version --- .github/scripts/validate_binaries.sh | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/.github/scripts/validate_binaries.sh b/.github/scripts/validate_binaries.sh index 0c01dbca2..acdcef5e5 100755 --- a/.github/scripts/validate_binaries.sh +++ b/.github/scripts/validate_binaries.sh @@ -12,6 +12,13 @@ else TEST_SUFFIX=" --package torchonly" fi + # if RELESE version is passed as parameter - install speific version + if [[ ! -z ${RELEASE_VERSION} ]]; then + INSTALLATION=${INSTALLATION/"torch "/"torch==${RELEASE_VERSION} "} + INSTALLATION=${INSTALLATION/"-y pytorch "/"-y pytorch==${RELEASE_VERSION} "} + INSTALLATION=${INSTALLATION/"::pytorch "/"::pytorch==${RELEASE_VERSION} "} + fi + export OLD_PATH=${PATH} # Workaround macos-arm64 runners. Issue: https://github.com/pytorch/test-infra/issues/4342 if [[ ${TARGET_OS} == 'macos-arm64' ]]; then