XLA pin downgrade to 20230825 (#5592)

Summary: The 20230826 version contains openxla/xla@3b8a539 whch regresses our LLaMA2 GSPMD training benchmark. Let's rollback to version before it. Test Plan: CI.
pytorch · Sep 18, 2023 · 9e1f87d · 9e1f87d
1 parent 5dd9108
commit 9e1f87d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/WORKSPACE b/WORKSPACE
@@ -43,9 +43,9 @@ http_archive(
         "//openxla_patches:gpu_race_condition.diff",
         "//openxla_patches:constexpr_return.diff",
     ],
-    strip_prefix = "xla-7a371ed44aba34f83d6d3d1159d2e6d0d327c603",
+    strip_prefix = "xla-97a5f819faf9ff793b7ba68ff1f31f74f9459c18",
     urls = [
-        "https://github.com/openxla/xla/archive/7a371ed44aba34f83d6d3d1159d2e6d0d327c603.tar.gz",
+        "https://github.com/openxla/xla/archive/97a5f819faf9ff793b7ba68ff1f31f74f9459c18.tar.gz",
     ],
 )
 

diff --git a/setup.py b/setup.py
@@ -72,7 +72,7 @@
 
 base_dir = os.path.dirname(os.path.abspath(__file__))
 
-_libtpu_version = '0.1.dev20230826'
+_libtpu_version = '0.1.dev20230825'
 _libtpu_storage_path = f'https://storage.googleapis.com/cloud-tpu-tpuvm-artifacts/wheels/libtpu-nightly/libtpu_nightly-{_libtpu_version}-py3-none-any.whl'