jax
b2d20891 - [JAX] Fail gracefully when an array with multiple shards is passed to make_array_from_single_device_arrays.

Commit

1 year ago

[JAX] Fail gracefully when an array with multiple shards is passed to make_array_from_single_device_arrays. Fix a crash when an exception is thrown during PyArray construction. PiperOrigin-RevId: 639760114

Author

hawkinsp

Committer

a-googler

Parents

fe35acd4

jax b2d20891 - [JAX] Fail gracefully when an array with multiple shards is passed to make_array_from_single_device_arrays.

jax
b2d20891 - [JAX] Fail gracefully when an array with multiple shards is passed to make_array_from_single_device_arrays.