From 7e8b8fb74fd62850faa1cf21ff884379620f9034 Mon Sep 17 00:00:00 2001
From: Yeounoh Chung <yeounoh@google.com>
Date: Mon, 25 Mar 2024 11:31:13 -0700
Subject: [PATCH] [r2.3] Update XLA pin  (#6797) (#6811)

---
 WORKSPACE | 4 ++--
 setup.py  | 2 +-
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/WORKSPACE b/WORKSPACE
index a1c4a3256bb..b04f6aa7d2d 100644
--- a/WORKSPACE
+++ b/WORKSPACE
@@ -50,9 +50,9 @@ http_archive(
         "//openxla_patches:gpu_race_condition.diff",
         "//openxla_patches:f16_abi_clang.diff",
     ],
-    strip_prefix = "xla-25c8a6781af6be51d3bc43a0953b07803ab761ea",
+    strip_prefix = "xla-01c52f2921c0eda9fdcc15d13227fbde08cbaca4",
     urls = [
-        "https://github.com/openxla/xla/archive/25c8a6781af6be51d3bc43a0953b07803ab761ea.tar.gz",
+        "https://github.com/openxla/xla/archive/01c52f2921c0eda9fdcc15d13227fbde08cbaca4.tar.gz",
     ],
 )
 
diff --git a/setup.py b/setup.py
index 16f14494f74..22b2e0b4956 100644
--- a/setup.py
+++ b/setup.py
@@ -64,7 +64,7 @@
 
 base_dir = os.path.dirname(os.path.abspath(__file__))
 
-_date = '20240320'
+_date = '20240322'
 _libtpu_version = f'0.1.dev{_date}'
 _libtpu_storage_path = f'https://storage.googleapis.com/cloud-tpu-tpuvm-artifacts/wheels/libtpu-nightly/libtpu_nightly-{_libtpu_version}-py3-none-any.whl'
 _jax_version = f'0.4.26.dev{_date}'