From be2f6a91ec9e0fc381327213018f2cf02d2dae56 Mon Sep 17 00:00:00 2001
From: Peter Hawkins <phawkins@google.com>
Date: Wed, 26 Jan 2022 22:19:11 +0000
Subject: [PATCH] Update XLA for jaxlib 0.1.76 release.

---
 CHANGELOG.md | 10 +++++++++-
 WORKSPACE    |  6 +++---
 2 files changed, 12 insertions(+), 4 deletions(-)

diff --git a/CHANGELOG.md b/CHANGELOG.md
index 8412f0948788..b39e3c90215f 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -16,17 +16,25 @@ PLEASE REMEMBER TO CHANGE THE '..main' WITH AN ACTUAL TAG in GITHUB LINK.
   * The `jax.jit(f).lower(...).compiler_ir(dialect='mhlo')` now returns an MLIR
     `ir.Module` object instead of its string representation.
 
-## jaxlib 0.1.76 (Unreleased)
+## jaxlib 0.1.77 (Unreleased)
+
+## jaxlib 0.1.76 (Jan 27, 2022)
 
 * New features
   * Includes precompiled SASS for NVidia compute capability 8.0 GPUS
     (e.g. A100). Removes precompiled SASS for compute capability 6.1 so as not
     to increase the number of compute capabilities: GPUs with compute capability
     6.1 can use the 6.0 SASS.
+  * With jaxlib 0.1.76, JAX uses the MHLO MLIR dialect as its primary target compiler IR
+    by default.
 * Breaking changes
   * Support for NumPy 1.18 has been dropped, per the
     [deprecation policy](https://jax.readthedocs.io/en/latest/deprecation.html).
     Please upgrade to a supported NumPy version.
+* Bug fixes
+  * Fixed a bug where apparently identical pytreedef objects constructed by different routes
+    do not compare as equal (#9066).
+  * The JAX jit cache requires two static arguments to have identical types for a cache hit (#9311).
 
 ## jax 0.2.27 (Jan 18 2022)
 * [GitHub commits](https://github.com/google/jax/compare/jax-v0.2.26...jax-v0.2.27).
diff --git a/WORKSPACE b/WORKSPACE
index 8872fb3fda5d..62d35899055c 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 = "org_tensorflow",
-    sha256 = "e62f0c241c090bf6753334bd1705dd1a528e3fedaf48e97a6a49e58f9de2b72f",
-    strip_prefix = "tensorflow-d8b3d5b22cb61b1a19e7d1c3524f1be92ee45ca6",
+    sha256 = "8d6604640c8092abf640ccfb7222f6f941ee7b409d30292f6261ce3c86e23fab",
+    strip_prefix = "tensorflow-e662039f98d2b6a0d8242887d0881bae04e2d590",
     urls = [
-        "https://github.com/tensorflow/tensorflow/archive/d8b3d5b22cb61b1a19e7d1c3524f1be92ee45ca6.tar.gz",
+        "https://github.com/tensorflow/tensorflow/archive/e662039f98d2b6a0d8242887d0881bae04e2d590.tar.gz",
     ],
 )