File tree 2 files changed +4
-4
lines changed
legacy_test/tests/pytorch/r2.7
2 files changed +4
-4
lines changed Original file line number Diff line number Diff line change @@ -18,7 +18,7 @@ local mixins = import 'templates/mixins.libsonnet';
18
18
local utils = import 'templates/utils.libsonnet' ;
19
19
local volumes = import 'templates/volumes.libsonnet' ;
20
20
21
- local rcVersion = 'rc4 ' ;
21
+ local rcVersion = 'rc5 ' ;
22
22
23
23
// make sure the vision commit aligns with upstream. E.g., for 2.7 release:
24
24
// https://github.com/pytorch/pytorch/blob/release/2.7/.github/ci_commit_pins/vision.txt.
Original file line number Diff line number Diff line change @@ -124,12 +124,12 @@ class R2_6(enum.Enum):
124
124
TORCH_XLA_REPO_BRANCH = "-b r2.6"
125
125
126
126
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"
129
129
TORCH = "torch==2.7.0"
130
130
TORCHVISION = "torchvision"
131
131
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"
133
133
TORCH_INDEX_CPU_URL = "https://download.pytorch.org/whl/test/cpu"
134
134
TORCH_INDEX_CUDA_URL = "https://download.pytorch.org/whl/test/cu126"
135
135
TORCH_REPO_BRANCH = "-b release/2.7"
You can’t perform that action at this time.
0 commit comments