jax
21f5f2d4 - [Pallas] Increase #rows when casting to x2.

Commit

345 days ago

[Pallas] Increase #rows when casting to x2. There is a bug in XLA on v5p. PiperOrigin-RevId: 736987667

References

Author

WindQAQ

Committer

Google-ML-Automation

Parents