From 9e1f87d9254132ba75675ea1571187a64405a27c Mon Sep 17 00:00:00 2001 From: Jiewen Tan Date: Mon, 18 Sep 2023 10:41:30 -0700 Subject: [PATCH] XLA pin downgrade to 20230825 (#5592) Summary: The 20230826 version contains https://github.com/openxla/xla/commit/3b8a5398ea9e479311d5c0c34bb1ae255551f9d4 whch regresses our LLaMA2 GSPMD training benchmark. Let's rollback to version before it. Test Plan: CI. --- WORKSPACE | 4 ++-- setup.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/WORKSPACE b/WORKSPACE index 1b51229132e..bf7d8c0c137 100644 --- 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 index 1d60a7776d9..f157fd72809 100644 --- 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'