6 Commits

Author	SHA1	Message	Date
Skye Wanderman-Milne	120125f3dd	Make pytest-xdist work on TPU and update Cloud TPU CI. ... This change also marks multiaccelerator test files in a way pytest can understand (if pytest is installed). By running single-device tests on a single TPU chip, running the test suite goes from 1hr 45m to 35m (both timings are running slow tests). I tried using bazel at first, which already supported parallel execution across TPU cores, but somehow it still takes 2h 20m! I'm not sure why it's so slow. It appears that bazel creates many new test processes over time, vs. pytest reuses the number of processes initially specified, and starting and stopping the TPU runtime takes a few seconds so that may be adding up. It also appears that single-process bazel is slower than single-process pytest, which I haven't looked into yet.	2022-11-18 22:05:13 +00:00
Peter Hawkins	ba557d5e1b	Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.". ... See https://opensource.google/documentation/reference/releasing/contributions#copyright for more details. PiperOrigin-RevId: 476167538	2022-09-22 12:27:19 -07:00
Peter Hawkins	541f762a31	Tolerate NumPy deprecation warnings when using older SciPy. ... Simply importing scipy 1.2.3 with NumPy 1.21.5 leads to deprecation warnings. Tolerate these in pytest.	2022-01-31 16:34:59 +00:00
Jake VanderPlas	cee3af580b	Testing: avoid global fixture for doctests	2021-04-08 12:11:14 -07:00
Jake VanderPlas	8e789c7380	Run doctest on all source files except jax2tf	2021-04-05 10:39:59 -07:00
Jake VanderPlas	af5b238007	Add empty conftest.py ... PiperOrigin-RevId: 366825504	2021-04-05 10:36:14 -07:00