Syncs hosts and empties async computation queue.
()
| 754 | |
| 755 | |
| 756 | def sync(): |
| 757 | """Syncs hosts and empties async computation queue.""" |
| 758 | x = jnp.ones([jax.local_device_count()]) |
| 759 | x = jax.device_get(jax.pmap(_sync, "i")(x)) |
| 760 | assert x[0] == jax.device_count() |
| 761 | |
| 762 | |
| 763 | def check_and_compile_patterns(patterns): |
nothing calls this directly
no outgoing calls
no test coverage detected