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

Remove unused return wrapper in annotate_function that creates a self reference cycle loop in python.

Browse Source 
PiperOrigin-RevId: 745099538
This commit is contained in:
jax authors 2025-04-08 05:24:34 -07:00
parent 5f33280ded
commit 73ecf0bb48
1 changed files with 0 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
jax/_src/profiler.py
View File

@ -332,7 +332,6 @@ def annotate_function(func: Callable, name: str | None = None,
def wrapper(*args, **kwargs):
with TraceAnnotation(name, **decorator_kwargs):
return func(*args, **kwargs)
return wrapper
return wrapper