Update WORKSPACE and setup.py in preparation for 0.3.15 jax/jaxlib re…

…lease
mbmccoy · Jul 14, 2022 · 9149c38 · 9149c38
1 parent 86ab8a8
commit 9149c38
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git 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 = "org_tensorflow",
-    sha256 = "634e5ee7fba57ba8e1e6af18c8521d9849d631248979f0fa4824e772ecabdb79",
-    strip_prefix = "tensorflow-6e406f0bd8d08be3fa43daa306e8565cdb29a546",
+    sha256 = "e3243973e04c182690e4a3b269b7daf1b94ee2e903afd06646d2486dcb79feae",
+    strip_prefix = "tensorflow-9ad5167145151eeb41eea01beb8d59441f47595d",
     urls = [
-        "https://github.com/tensorflow/tensorflow/archive/6e406f0bd8d08be3fa43daa306e8565cdb29a546.tar.gz",
+        "https://github.com/tensorflow/tensorflow/archive/9ad5167145151eeb41eea01beb8d59441f47595d.tar.gz",
     ],
 )
 

diff --git a/setup.py b/setup.py
@@ -14,14 +14,14 @@
 
 from setuptools import setup, find_packages
 
-_current_jaxlib_version = '0.3.14'
+_current_jaxlib_version = '0.3.15'
 # The following should be updated with each new jaxlib release.
 _latest_jaxlib_version_on_pypi = '0.3.14'
 _available_cuda_versions = ['11']
 _default_cuda_version = '11'
 _available_cudnn_versions = ['82', '805']
 _default_cudnn_version = '82'
-_libtpu_version = '0.1.dev20220627'
+_libtpu_version = '0.1.dev20220714'
 
 _dct = {}
 with open('jax/version.py') as f: