From bd166e1d99f78ee2bad833ab1be1b40f9719b23d Mon Sep 17 00:00:00 2001 From: Matthew Johnson Date: Mon, 1 Jul 2024 22:21:10 +0000 Subject: [PATCH] add more info to xla_computation deprecation warning --- jax/__init__.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/jax/__init__.py b/jax/__init__.py index df96b98cd..cedbec4b8 100644 --- a/jax/__init__.py +++ b/jax/__init__.py @@ -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 ), }