From 93e19e8a8593fbbba6ac350285bb1c282021d886 Mon Sep 17 00:00:00 2001
From: Pei Zhang <piz@google.com>
Date: Wed, 17 Jan 2024 10:49:39 -0800
Subject: [PATCH] update ptxla test to use rc8 tag build (#1043)

---
 tests/pytorch/r2.2/common.libsonnet   | 4 ++--
 tests/pytorch/r2.2/sd-model.libsonnet | 2 +-
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/tests/pytorch/r2.2/common.libsonnet b/tests/pytorch/r2.2/common.libsonnet
index f36688156..5e2b10f20 100644
--- a/tests/pytorch/r2.2/common.libsonnet
+++ b/tests/pytorch/r2.2/common.libsonnet
@@ -24,7 +24,7 @@ local volumes = import 'templates/volumes.libsonnet';
     tpuSettings+: {
       softwareVersion: 'pytorch-2.2',
     },
-    imageTag: 'r2.2.0-rc6_3.10',
+    imageTag: 'r2.2.0-rc8_3.10',
   },
   PyTorchTest:: common.PyTorchTest + r2_2 {
     local config = self,
@@ -97,7 +97,7 @@ local volumes = import 'templates/volumes.libsonnet';
         # for huggingface tests
         sudo apt install -y libsndfile-dev
         pip3 install torch==2.2.0 torchvision torchaudio --index-url https://download.pytorch.org/whl/test/cpu
-        pip install https://storage.googleapis.com/pytorch-xla-releases/wheels/tpuvm/torch_xla-2.2.0rc6-cp310-cp310-manylinux_2_28_x86_64.whl
+        pip install https://storage.googleapis.com/pytorch-xla-releases/wheels/tpuvm/torch_xla-2.2.0rc8-cp310-cp310-manylinux_2_28_x86_64.whl
         pip install torch_xla[tpu] -f https://storage.googleapis.com/libtpu-releases/index.html
         pip3 install pillow
         git clone --depth=1 -b release/2.2 https://github.com/pytorch/pytorch.git
diff --git a/tests/pytorch/r2.2/sd-model.libsonnet b/tests/pytorch/r2.2/sd-model.libsonnet
index 4e7854c40..0d6dbea6f 100644
--- a/tests/pytorch/r2.2/sd-model.libsonnet
+++ b/tests/pytorch/r2.2/sd-model.libsonnet
@@ -65,7 +65,7 @@ local utils = import 'templates/utils.libsonnet';
         pip uninstall -y torch torchvision
         pip3 install --user --pre torch torchvision --index-url https://download.pytorch.org/whl/test/cpu
         pip install --user \
-          'torch_xla[tpuvm] @ https://storage.googleapis.com/pytorch-xla-releases/wheels/tpuvm/torch_xla-2.2.0rc6-cp310-cp310-manylinux_2_28_x86_64.whl'
+          'torch_xla[tpuvm] @ https://storage.googleapis.com/pytorch-xla-releases/wheels/tpuvm/torch_xla-2.2.0rc8-cp310-cp310-manylinux_2_28_x86_64.whl'
 
         # Setup data
         wget -nv https://s3.amazonaws.com/fast-ai-imageclas/imagenette2.tgz