jax
dfda948f - [XLA:Python] Fail with an AttributeError if __cuda_array_interface__ is called on a sharded array.

Commit

2 years ago

[XLA:Python] Fail with an AttributeError if __cuda_array_interface__ is called on a sharded array. Fixes https://github.com/google/jax/issues/19134 PiperOrigin-RevId: 600570354

Author

hawkinsp

Committer

a-googler

Parents

46f796b3

jax dfda948f - [XLA:Python] Fail with an AttributeError if __cuda_array_interface__ is called on a sharded array.

jax
dfda948f - [XLA:Python] Fail with an AttributeError if __cuda_array_interface__ is called on a sharded array.