Skip to content

Commit 3723ee7

Browse files
authored
Validation scripts, install using version (#1633)
1 parent b92da8c commit 3723ee7

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

.github/scripts/validate_binaries.sh

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,13 @@ else
1212
TEST_SUFFIX=" --package torchonly"
1313
fi
1414

15+
# if RELESE version is passed as parameter - install speific version
16+
if [[ ! -z ${RELEASE_VERSION} ]]; then
17+
INSTALLATION=${INSTALLATION/"torch "/"torch==${RELEASE_VERSION} "}
18+
INSTALLATION=${INSTALLATION/"-y pytorch "/"-y pytorch==${RELEASE_VERSION} "}
19+
INSTALLATION=${INSTALLATION/"::pytorch "/"::pytorch==${RELEASE_VERSION} "}
20+
fi
21+
1522
export OLD_PATH=${PATH}
1623
# Workaround macos-arm64 runners. Issue: https://github.com/pytorch/test-infra/issues/4342
1724
if [[ ${TARGET_OS} == 'macos-arm64' ]]; then

0 commit comments

Comments
 (0)