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

Merge pull request #22222 from mattjj:xla-computation-deprecation

Browse Source 
PiperOrigin-RevId: 648719949
This commit is contained in:
jax authors 2024-07-02 08:03:23 -07:00
commit 42724ebc73
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.", "jax.clear_backends is deprecated.",
_deprecated_clear_backends _deprecated_clear_backends
), ),
# Added Jun 16, 2024
"xla_computation": ( "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 _deprecated_xla_computation
), ),
} }