mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-17 12:26:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

add more info to xla_computation deprecation warning

Browse Source
This commit is contained in:
Matthew Johnson 2024-07-01 22:21:10 +00:00
parent a2a5068e5e
commit bd166e1d99
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
jax/__init__.py
View File

@ -226,8 +226,12 @@ _deprecations = {
"jax.clear_backends is deprecated.",
_deprecated_clear_backends
),
# Added Jun 16, 2024
"xla_computation": (
"jax.xla_computation is deprecated. Please use the AOT APIs.",
"jax.xla_computation is deprecated. Please use the AOT APIs; see "
"https://jax.readthedocs.io/en/latest/aot.html. For example, replace "
"xla_computation(f)(*xs) with jit(f).lower(*xs).compiler_ir('hlo'). See "
"CHANGELOG.md for 0.4.30 for more examples.",
_deprecated_xla_computation
),
}