Skip to main content

Browse the archive

Revision 4c423c36d0b4f0e05c29db507bd6bedd5e0f9e75 authored by Peter Hawkins on 18 January 2022, 18:41:43 UTC, committed by jax authors on 18 January 2022, 18:42:14 UTC 
Speed up check_jaxpr().
 
(check_jaxpr() is only used when debugging.)

Don't eagerly pretty print jaxprs: only do so if we are going to raise an error.
Don't eagerly form error messages. Delete typecheck_assert.

PiperOrigin-RevId: 422594126
1 parent e30b96c
Raw File
pytest.ini 
[pytest]
filterwarnings =
    error
    ignore:No GPU/TPU found, falling back to CPU.:UserWarning
    ignore:outfeed_receiver is unnecessary and deprecated:DeprecationWarning
    # xmap
    ignore:xmap is an experimental feature and probably has bugs!
    # The rest are for experimental/jax_to_tf
    ignore:the imp module is deprecated in favour of importlib.*:DeprecationWarning
    ignore:can't resolve package from __spec__ or __package__:ImportWarning
    ignore:Using or importing the ABCs.*:DeprecationWarning
    # jax2tf tests due to mix of JAX and TF
    ignore:numpy.ufunc size changed
    ignore:.*experimental feature
    ignore:index.*is deprecated.*:DeprecationWarning
    ignore:jax.experimental.* is deprecated, import jax.example_libraries.* instead:FutureWarning
    # numpy uses distutils which is deprecated
    ignore:The distutils.* is deprecated.*:DeprecationWarning
doctest_optionflags = NUMBER NORMALIZE_WHITESPACE
addopts = --doctest-glob="*.rst"
back to top