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

check path prefixes using os.path instead of string comparisons

Browse Source
This commit is contained in:
Roy Frostig 2020-08-18 17:03:43 -07:00 committed by Roy Frostig
parent d778a6d074
commit 8cc9579ca0
1 changed files with 10 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
jax/traceback_util.py
View File

@ -32,9 +32,17 @@ _jax_message_append = (
'The following is the original exception that occurred, unmodified.\n'
'\n--------------------')
def path_starts_with(path, path_prefix):
path = os.path.abspath(path)
path_prefix = os.path.abspath(path_prefix)
common = os.path.commonpath([path, path_prefix])
return os.path.samefile(common, path_prefix)
def include_frame(f):
return (not f.f_code.co_filename.startswith(_jax_path) or
any(f.f_code.co_filename.startswith(path) for path in _include_paths))
return (
not path_starts_with(f.f_code.co_filename, _jax_path) or
any(path_starts_with(f.f_code.co_filename, path)
for path in _include_paths))
# When scanning stack traces, we might encounter frames from cpython that are
# removed from printed stack traces, such as frames from parts of importlib. We