mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-14 10:56:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Finalize deprecation of some symbols from jax.lib.xla_client

Browse Source
This commit is contained in:
Jake VanderPlas 2024-12-23 09:23:06 -08:00
parent 108762379d
commit 40fe4b8797
2 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
CHANGELOG.md
View File

@ -34,6 +34,9 @@ When releasing, please add the new-release-boilerplate to docs/pallas/CHANGELOG.
* Deletions
* `jax_enable_memories` flag has been deleted and the behavior of that flag
is on by default.
* From `jax.lib.xla_client`, the previously-deprecated `Device` and
`XlaRuntimeError` symbols have been removed; instead use `jax.Device`
and `jax.errors.JaxRuntimeError` respectively.
## jax 0.4.38 (Dec 17, 2024)

6
jax/lib/xla_client.py
View File

@ -36,17 +36,17 @@ _deprecations = {
"jax.lib.xla_client.bfloat16 was removed in JAX v0.4.38; use ml_dtypes.bfloat16.",
None,
),
# Added Sep 26 2024
# Finalized 2024-12-23; remove after 2024-03-23
"Device": (
"jax.lib.xla_client.Device is deprecated; use jax.Device instead.",
_xc.Device,
None,
),
"XlaRuntimeError": (
(
"jax.lib.xla_client.XlaRuntimeError is deprecated; use"
" jax.errors.JaxRuntimeError."
),
_xc.XlaRuntimeError,
None,
),
# Added Oct 10 2024
"FftType": (