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

improve leak checker flag description

Browse Source
This commit is contained in:
Matthew Johnson 2021-01-23 14:17:22 -08:00
parent 9787894d94
commit a7bfebe4bc
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
jax/config.py
View File

@ -174,5 +174,7 @@ flags.DEFINE_integer(
flags.DEFINE_bool(
'jax_check_tracer_leaks',
bool_env('JAX_CHECK_TRACER_LEAKS', False),
help='Turn on checking for leaked tracers as soon as a trace completes.'
help=('Turn on checking for leaked tracers as soon as a trace completes. '
'Enabling leak checking may have performance impacts: some caching '
'is disabled, and other overheads may be added.'),
)