Validation scripts, install using version (#1633)

pytorch · Dec 6, 2023 · 3723ee7 · 3723ee7
1 parent b92da8c
commit 3723ee7
Showing 1 changed file with 7 additions and 0 deletions.
diff --git 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