jax
df6a0fbd - [Pallas TPU] Raise `NotImplementedError` when casting to 64-bit integer

Commit

1 year ago

[Pallas TPU] Raise `NotImplementedError` when casting to 64-bit integer This fixes https://github.com/jax-ml/jax/issues/23988 PiperOrigin-RevId: 684121559

Author

ayaka14732

Committer

Google-ML-Automation

Parents

f96c5661

jax df6a0fbd - [Pallas TPU] Raise `NotImplementedError` when casting to 64-bit integer

jax
df6a0fbd - [Pallas TPU] Raise `NotImplementedError` when casting to 64-bit integer