From e9cac5eb4739e2a1635e34d56011ee758d08173f Mon Sep 17 00:00:00 2001 From: Yash Katariya Date: Mon, 27 Mar 2023 11:44:19 -0700 Subject: [PATCH] Prepare for jax and jaxlib 0.4.7 release PiperOrigin-RevId: 519785176 --- WORKSPACE | 2 +- setup.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/WORKSPACE b/WORKSPACE index da8723791bfa..ab7ada26af73 100644 --- a/WORKSPACE +++ b/WORKSPACE @@ -7,7 +7,7 @@ load("@bazel_tools//tools/build_defs/repo:http.bzl", "http_archive") # and update the sha256 with the result. http_archive( name = "xla", - sha256 = "9f39af4d81d2c8bd52b47f4ef37dfd6642c6950112e4d9d3d95ae19982c46eba", + sha256 = "e1f99fbc3d149a8a2f83986f6ed4c0895d2901c0faf41fa6b26efc58948ac82e", strip_prefix = "xla-0f31407ee498e6dba242d03f8d382ebcfcc61790", urls = [ "https://github.com/openxla/xla/archive/0f31407ee498e6dba242d03f8d382ebcfcc61790.tar.gz", diff --git a/setup.py b/setup.py index e2e3b4679790..8f16423db820 100644 --- a/setup.py +++ b/setup.py @@ -19,13 +19,13 @@ from setuptools import setup, find_packages -_current_jaxlib_version = '0.4.6' +_current_jaxlib_version = '0.4.7' # The following should be updated with each new jaxlib release. _latest_jaxlib_version_on_pypi = '0.4.6' _available_cuda11_cudnn_versions = ['82', '86'] _default_cuda11_cudnn_version = '86' _default_cuda12_cudnn_version = '88' -_libtpu_version = '0.1.dev20230309' +_libtpu_version = '0.1.dev20230327' _dct = {} with open('jax/version.py', encoding='utf-8') as f: