File tree Expand file tree Collapse file tree 2 files changed +13
-4
lines changed Expand file tree Collapse file tree 2 files changed +13
-4
lines changed Original file line number Diff line number Diff line change 10
10
11
11
if [[ ${TORCH_ONLY} == ' true' ]]; then
12
12
TEST_SUFFIX=" --package torchonly"
13
- pip3 install torch${RELEASE_SUFFIX}
13
+ pip3 install --pre torch${RELEASE_SUFFIX} --extra-index-url " https://download.pytorch.org/whl/ ${MATRIX_CHANNEL} / ${MATRIX_DESIRED_CUDA} _pypi_cudnn "
14
14
else
15
- pip3 install torch${RELEASE_SUFFIX} torchvision torchaudio
15
+ if [[ ${MATRIX_CHANNEL} != " release" ]]; then
16
+ pip3 install --pre torch${RELEASE_SUFFIX} --extra-index-url " https://download.pytorch.org/whl/${MATRIX_CHANNEL} /${MATRIX_DESIRED_CUDA} _pypi_cudnn"
17
+ pip3 install --pre torchvision torchaudio --extra-index-url " https://download.pytorch.org/whl/${MATRIX_CHANNEL} /${MATRIX_DESIRED_CUDA} "
18
+ else
19
+ pip3 install torch${RELEASE_SUFFIX} torchvision torchaudio
20
+ fi
16
21
fi
17
22
18
23
python ./test/smoke_test/smoke_test.py ${TEST_SUFFIX} --runtime-error-check disabled
Original file line number Diff line number Diff line change @@ -88,11 +88,15 @@ jobs:
88
88
export TARGET_OS="linux"
89
89
eval "$(conda shell.bash hook)"
90
90
printf '%s\n' ${{ toJson(inputs.release-matrix) }} > release_matrix.json
91
+ cat release_matrix.json
91
92
92
93
# Special case PyPi installation package. And Install of PyPi package via poetry
93
- if [[ ${MATRIX_PACKAGE_TYPE} == "manywheel" && ${MATRIX_GPU_ARCH_VERSION} == "12.1" && ${MATRIX_CHANNEL} == "release" ]]; then
94
+ if [[ ${MATRIX_PACKAGE_TYPE} == "manywheel" && ${MATRIX_GPU_ARCH_VERSION} == "12.1" ]]; then
94
95
source ./.github/scripts/validate_pipy.sh
95
- source ./.github/scripts/validate_poetry.sh
96
+
97
+ if [[ ${MATRIX_CHANNEL} == "release" ]]; then
98
+ source ./.github/scripts/validate_poetry.sh
99
+ fi
96
100
fi
97
101
98
102
# Standart case: Validate binaries
You can’t perform that action at this time.
0 commit comments