mirror of
https://github.com/ROCm/jax.git
synced 2025-04-15 19:36:06 +00:00
Update XLA for jaxlib 0.1.76 release.
This commit is contained in:
parent
6bd9292b3d
commit
be2f6a91ec
10
CHANGELOG.md
10
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).
|
||||
|
@ -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",
|
||||
],
|
||||
)
|
||||
|
||||
|
Loading…
x
Reference in New Issue
Block a user