Skip to content

Commit 4757a6d

Browse files
authored
Test against torch RC instead of nightly (#6080)
1 parent ea6f139 commit 4757a6d

File tree

3 files changed

+11
-19
lines changed

3 files changed

+11
-19
lines changed

.circleci/config.yml

Lines changed: 4 additions & 8 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

.circleci/config.yml.in

Lines changed: 4 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -39,11 +39,7 @@ commands:
3939
- run:
4040
name: adding UPLOAD_CHANNEL to BASH_ENV
4141
command: |
42-
our_upload_channel=nightly
43-
# On tags upload to test instead
44-
if [[ -n "${CIRCLE_TAG}" ]]; then
45-
our_upload_channel=test
46-
fi
42+
our_upload_channel=test
4743
echo "export UPLOAD_CHANNEL=${our_upload_channel}" >> ${BASH_ENV}
4844

4945
brew_update:
@@ -146,7 +142,7 @@ commands:
146142
default: true
147143
steps:
148144
- pip_install:
149-
args: --pre torch --extra-index-url https://download.pytorch.org/whl/nightly/cpu
145+
args: --pre torch --extra-index-url https://download.pytorch.org/whl/test/cpu/torch_test.html
150146
descr: Install PyTorch from nightly releases
151147
- pip_install:
152148
args: --no-build-isolation <<# parameters.editable >> --editable <</ parameters.editable >> .
@@ -637,7 +633,7 @@ jobs:
637633
set -x
638634
source /usr/local/etc/profile.d/conda.sh && conda activate python${PYTHON_VERSION}
639635
- pip_install:
640-
args: $(ls ~/workspace/torchvision*.whl) --pre -f https://download.pytorch.org/whl/nightly/torch_nightly.html
636+
args: $(ls ~/workspace/torchvision*.whl) --pre -f https://download.pytorch.org/whl/test/torch_test.html
641637
- run:
642638
name: smoke test
643639
command: |
@@ -706,7 +702,7 @@ jobs:
706702
conda create -yn python${PYTHON_VERSION} python=${PYTHON_VERSION}
707703
conda activate python${PYTHON_VERSION}
708704
- pip_install:
709-
args: $(ls ~/workspace/torchvision*.whl) --pre -f https://download.pytorch.org/whl/nightly/torch_nightly.html
705+
args: $(ls ~/workspace/torchvision*.whl) --pre -f https://download.pytorch.org/whl/test/torch_test.html
710706
- run:
711707
name: smoke test
712708
command: |

packaging/pkg_helpers.bash

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -184,7 +184,7 @@ setup_pip_pytorch_version() {
184184
if [[ -z "$PYTORCH_VERSION" ]]; then
185185
# Install latest prerelease version of torch, per our nightlies, consistent
186186
# with the requested cuda version
187-
pip_install --pre torch -f "https://download.pytorch.org/whl/nightly/${WHEEL_DIR}torch_nightly.html"
187+
pip_install --pre torch -f "https://download.pytorch.org/whl/test/${WHEEL_DIR}torch_test.html"
188188
if [[ "$CUDA_VERSION" == "cpu" ]]; then
189189
# CUDA and CPU are ABI compatible on the CPU-only parts, so strip
190190
# in this case
@@ -205,8 +205,8 @@ setup_pip_pytorch_version() {
205205
# You MUST have populated PYTORCH_VERSION_SUFFIX before hand.
206206
setup_conda_pytorch_constraint() {
207207
if [[ -z "$PYTORCH_VERSION" ]]; then
208-
export CONDA_CHANNEL_FLAGS="${CONDA_CHANNEL_FLAGS} -c pytorch-nightly -c pytorch"
209-
export PYTORCH_VERSION="$(conda search --json 'pytorch[channel=pytorch-nightly]' | \
208+
export CONDA_CHANNEL_FLAGS="${CONDA_CHANNEL_FLAGS} -c pytorch-test -c pytorch"
209+
export PYTORCH_VERSION="$(conda search --json 'pytorch[channel=pytorch-test]' | \
210210
python -c "import os, sys, json, re; cuver = os.environ.get('CU_VERSION'); \
211211
cuver_1 = cuver.replace('cu', 'cuda') if cuver != 'cpu' else cuver; \
212212
cuver_2 = (cuver[:-1] + '.' + cuver[-1]).replace('cu', 'cuda') if cuver != 'cpu' else cuver; \

0 commit comments

Comments
 (0)