From 5bcd9dcc4645be894a278e2323e668ce334234a6 Mon Sep 17 00:00:00 2001
From: Skye Wanderman-Milne <skyewm@google.com>
Date: Tue, 9 May 2023 14:49:54 -0700
Subject: [PATCH] Update WORKSPACE and setup.py in preparation for jax/jaxlib
 0.4.9 release, take 2

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

diff --git a/WORKSPACE b/WORKSPACE
index 292841cbdbd6..627e1de794f6 100644
--- a/WORKSPACE
+++ b/WORKSPACE
@@ -7,10 +7,10 @@ load("@bazel_tools//tools/build_defs/repo:http.bzl", "http_archive")
 #    and update the sha256 with the result.
 http_archive(
     name = "xla",
-    sha256 = "0e32c717ec6091ac9d2a620ca895a90f1b996657a36fca952ade4ccda54e5eea",
-    strip_prefix = "xla-48b5796215744b2de022250720731c782039c39d",
+    sha256 = "3d9ca2c8cd81b820b63052409867a81102c42bb1382b971f6402bad21288df52",
+    strip_prefix = "xla-dad64948516e3672b3e2518945831a70b5e90b81",
     urls = [
-        "https://github.com/openxla/xla/archive/48b5796215744b2de022250720731c782039c39d.tar.gz",
+        "https://github.com/openxla/xla/archive/dad64948516e3672b3e2518945831a70b5e90b81.tar.gz",
     ],
 )
 
diff --git a/setup.py b/setup.py
index 0b7d798124cf..4b9a3bde8383 100644
--- a/setup.py
+++ b/setup.py
@@ -19,13 +19,13 @@
 
 from setuptools import setup, find_packages
 
-_current_jaxlib_version = '0.4.7'
+_current_jaxlib_version = '0.4.9'
 # The following should be updated with each new jaxlib release.
 _latest_jaxlib_version_on_pypi = '0.4.7'
 _available_cuda11_cudnn_versions = ['82', '86']
 _default_cuda11_cudnn_version = '86'
 _default_cuda12_cudnn_version = '88'
-_libtpu_version = '0.1.dev20230327'
+_libtpu_version = '0.1.dev20230509'
 
 _dct = {}
 with open('jax/version.py', encoding='utf-8') as f: