Skip to content

Commit e32a00b

Browse files
authored
enable rc5 test (#681)
1 parent d9079df commit e32a00b

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

dags/legacy_test/tests/pytorch/r2.7/common.libsonnet

+1-1
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@ local mixins = import 'templates/mixins.libsonnet';
1818
local utils = import 'templates/utils.libsonnet';
1919
local volumes = import 'templates/volumes.libsonnet';
2020

21-
local rcVersion = 'rc4';
21+
local rcVersion = 'rc5';
2222

2323
// make sure the vision commit aligns with upstream. E.g., for 2.7 release:
2424
// https://github.com/pytorch/pytorch/blob/release/2.7/.github/ci_commit_pins/vision.txt.

dags/pytorch_xla/configs/pytorchxla_torchbench_config.py

+3-3
Original file line numberDiff line numberDiff line change
@@ -124,12 +124,12 @@ class R2_6(enum.Enum):
124124
TORCH_XLA_REPO_BRANCH = "-b r2.6"
125125

126126
class R2_7(enum.Enum):
127-
TORCH_XLA_TPU_WHEEL = "https://storage.googleapis.com/pytorch-xla-releases/wheels/tpuvm/torch_xla-2.7.0rc4-cp311-cp311-manylinux_2_28_x86_64.whl"
128-
TORCH_XLA_CUDA_WHEEL = "https://storage.googleapis.com/pytorch-xla-releases/wheels/cuda/12.6/torch_xla-2.7.0rc4-cp310-cp310-linux_x86_64.whl"
127+
TORCH_XLA_TPU_WHEEL = "https://storage.googleapis.com/pytorch-xla-releases/wheels/tpuvm/torch_xla-2.7.0rc5-cp311-cp311-manylinux_2_28_x86_64.whl"
128+
TORCH_XLA_CUDA_WHEEL = "https://storage.googleapis.com/pytorch-xla-releases/wheels/cuda/12.6/torch_xla-2.7.0rc5-cp310-cp310-linux_x86_64.whl"
129129
TORCH = "torch==2.7.0"
130130
TORCHVISION = "torchvision"
131131
TORCHAUDIO = "torchaudio"
132-
TORCH_XLA_GPU_DOCKER = "us-central1-docker.pkg.dev/tpu-pytorch-releases/docker/xla:r2.7.0-rc4_3.10_cuda_12.6"
132+
TORCH_XLA_GPU_DOCKER = "us-central1-docker.pkg.dev/tpu-pytorch-releases/docker/xla:r2.7.0-rc5_3.10_cuda_12.6"
133133
TORCH_INDEX_CPU_URL = "https://download.pytorch.org/whl/test/cpu"
134134
TORCH_INDEX_CUDA_URL = "https://download.pytorch.org/whl/test/cu126"
135135
TORCH_REPO_BRANCH = "-b release/2.7"

0 commit comments

Comments
 (0)