Reverts d3259a5

PiperOrigin-RevId: 861384560

Author: nitins17

MODULE.bazel

@@ -26,9 +26,9 @@ archive_override(
 bazel_dep(name = "xla")
 archive_override(
     module_name = "xla",
-    integrity = "sha256-VnnIS2ar7FYEVzKCQnJ693koAIkmEOasDzkgoGjr7ls=",
-    strip_prefix = "xla-719494360c30b789370b6a7e8d1fa9be484b34e0",
-    urls = ["https://github.com/openxla/xla/archive/719494360c30b789370b6a7e8d1fa9be484b34e0.tar.gz"],
+    integrity = "sha256-FmmcrZgng83LaOHnuYpTmouqcQdCh5jbt0tnUQQdj9g=",
+    strip_prefix = "xla-ed953c01bb51f95a36abd907d1a64295feef16fc",
+    urls = ["https://github.com/openxla/xla/archive/ed953c01bb51f95a36abd907d1a64295feef16fc.tar.gz"],
 )
 
 # TODO: upstream, otherwise we have to duplicate the patches in jax

third_party/xla/revision.bzl

@@ -21,5 +21,5 @@
 #    and update XLA_SHA256 with the result.
 
 # buildifier: disable=module-docstring
-XLA_COMMIT = "719494360c30b789370b6a7e8d1fa9be484b34e0"
-XLA_SHA256 = "5679c84b66abec560457328242727af7792800892610e6ac0f3920a068ebee5b"
+XLA_COMMIT = "a9cb3c90c30895103ffbb4fc231b9930d5c61b91"
+XLA_SHA256 = "d6c2982852844adbaf643f58398cc3d5fd159d4c2227e4a78c58dde7fda0688f"