jax
6625a2b3 - Update Eigh kernel on GPU to use 64-bit interface when it is available.

Commit

1 year ago

Update Eigh kernel on GPU to use 64-bit interface when it is available. Part of https://github.com/jax-ml/jax/issues/23413 PiperOrigin-RevId: 684546802

References

#24216 - Update Eigh kernel on GPU to use 64-bit interface when it is available.

Author

dfm

Committer

Google-ML-Automation

Parents

9d44d723

jax 6625a2b3 - Update Eigh kernel on GPU to use 64-bit interface when it is available.

jax
6625a2b3 - Update Eigh kernel on GPU to use 64-bit interface when it is available.