Skip to content

Commit 3c2d43c

Browse files
authored
Validation pypi torchonly (#1505)
1 parent 8b39596 commit 3c2d43c

File tree

1 file changed

+10
-4
lines changed

1 file changed

+10
-4
lines changed

.github/scripts/validate_pipy.sh

Lines changed: 10 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,12 +1,18 @@
11
conda create -yp ${ENV_NAME}_pypi python=${MATRIX_PYTHON_VERSION} numpy ffmpeg
22

3-
if [[ ${MATRIX_CHANNEL} != "release" ]]; then
3+
TEST_SUFFIX=""
4+
if [[ ${TORCH_ONLY} == 'true' ]]; then
5+
TEST_SUFFIX=" --package torchonly"
46
conda run -p ${ENV_NAME}_pypi pip3 install --pre torch --index-url "https://download.pytorch.org/whl/${MATRIX_CHANNEL}/${MATRIX_DESIRED_CUDA}_pypi_cudnn"
5-
conda run -p ${ENV_NAME}_pypi pip3 install --pre torchvision torchaudio --index-url "https://download.pytorch.org/whl/${MATRIX_CHANNEL}/${MATRIX_DESIRED_CUDA}"
67
else
7-
conda run -p ${ENV_NAME}_pypi pip3 install torch torchvision torchaudio
8+
if [[ ${MATRIX_CHANNEL} != "release" ]]; then
9+
conda run -p ${ENV_NAME}_pypi pip3 install --pre torch --index-url "https://download.pytorch.org/whl/${MATRIX_CHANNEL}/${MATRIX_DESIRED_CUDA}_pypi_cudnn"
10+
conda run -p ${ENV_NAME}_pypi pip3 install --pre torchvision torchaudio --index-url "https://download.pytorch.org/whl/${MATRIX_CHANNEL}/${MATRIX_DESIRED_CUDA}"
11+
else
12+
conda run -p ${ENV_NAME}_pypi pip3 install torch torchvision torchaudio
13+
fi
814
fi
915

10-
conda run -p ${ENV_NAME}_pypi python ./test/smoke_test/smoke_test.py
16+
conda run -p ${ENV_NAME}_pypi python ./test/smoke_test/smoke_test.py ${TEST_SUFFIX} --runtime-error-check disabled
1117
conda deactivate
1218
conda env remove -p ${ENV_NAME}_pypi

0 commit comments

Comments
 (0)