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 #23726 from hawkinsp:debug

Browse Source 
PiperOrigin-RevId: 676030839
This commit is contained in:
jax authors 2024-09-18 10:00:03 -07:00
commit cd04d0f32e
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
jax/_src/debugger/core.py
View File

@ -112,6 +112,11 @@ class DebuggerFrame:
# then we subtract it off from the `lineno` and don't need to subtract 1
# since both start and lineno are 1-indexed.
offset = frame_info.lineno - max(start, 1)
if offset >= len(source):
# Sometimes we don't get a valid source/offset pair. This seems to
# happen sometimes when code uses eval(). If that happens, give up.
source = []
offset = None
except OSError:
source = []
offset = None