15 Commits

Author SHA1 Message Date
Peter Hawkins
89a6b6d1a0 Fix stack trace filtering for paths that do not exist.
PiperOrigin-RevId: 371521548
2021-05-01 16:13:07 -07:00
Peter Hawkins
a80aab0a38 Fix cloudpickle breakage.
PiperOrigin-RevId: 371517484
2021-05-01 14:57:41 -07:00
Peter Hawkins
e8c340623c [JAX] Switch the order of the filtered and unfiltered stack traces in exceptions.
After this change, the filtered stack trace is attached to the main exception, and the unfiltered stack trace becomes a __cause__ exception.

PiperOrigin-RevId: 371509766
2021-05-01 12:41:37 -07:00
Peter Hawkins
ccc423d680 [JAX] Enable filtered tracebacks on Python 3.6.
[XLA:Python] Add support for converting a fast-traceback into a Python exception traceback.

Add a helper for building traceback objects on Python 3.6. On Python 3.7+ this can be done by calling the traceback type, and we can in essence backport that implementation to Python 3.6.

Consolidate the py_traceback and traceback modules.

PiperOrigin-RevId: 371193212
2021-04-29 13:37:21 -07:00
Roy Frostig
2fc2ff409a tiny change for source sync
PiperOrigin-RevId: 361249206
2021-03-05 16:32:28 -08:00
Roy Frostig
306fb0bf3d Merge pull request #5958 from jakevdp:glossary
PiperOrigin-RevId: 361240568
2021-03-05 15:54:01 -08:00
Roy Frostig
1c20e70839 test source sync
PiperOrigin-RevId: 359687592
2021-02-25 22:07:03 -08:00
Matthew Johnson
eb61395e6b Merge pull request #5823 from jakevdp:jax-101
PiperOrigin-RevId: 359679823
2021-02-25 21:08:04 -08:00
Peter Hawkins
3ac809ede3 [JAX] Move jax.util to jax._src_util.
PiperOrigin-RevId: 351234602
2021-01-11 14:21:07 -08:00
Roy Frostig
897916ffc4 test source sync
PiperOrigin-RevId: 350663594
2021-01-07 16:45:11 -08:00
Roy Frostig
432ef31342 tiny change for source sync
PiperOrigin-RevId: 350379734
2021-01-06 10:23:21 -08:00
Roy Frostig
c699329c54 tiny change to test CI
PiperOrigin-RevId: 348870843
2020-12-23 17:58:31 -08:00
Roy Frostig
6cddc2fa77 tiny change for source sync
PiperOrigin-RevId: 348839503
2020-12-23 13:24:58 -08:00
Cloud Han
a6acce58e0 Build on Windows
1. Build on Windows

2. Fix OverflowError

    When calling `key = random.PRNGKey(0)` OverflowError: Python int too
    large to convert to C long for casting value 4294967295 (0xFFFFFFFF)
    from python int to int32.

3. fix file path in regex of errors_test

4. handle ValueError of os.path.commonpath
2020-11-19 23:33:06 +08:00
Peter Hawkins
81b6cd29ff [JAX] Move traceback_util.py into jax._src.
traceback_util is a JAX-private API.

PiperOrigin-RevId: 340659195
2020-11-04 09:02:59 -08:00