jax
e7545eca - [Pallas] Better error message for grid/dimension semantics mismatch.

Commit

47 days ago

[Pallas] Better error message for grid/dimension semantics mismatch. PiperOrigin-RevId: 827516689

References

#31381 - Remove old ROCm build code

#33049 - [Pallas] Better error message for grid/dimension semantics mismatch.

#33157 - Resolve undefined behavior in bitshift unit test

#33186 - Make nvidia version data optional for ROCm builds

Author

justinjfu

justinjfu

Committer

Google-ML-Automation

Google-ML-Automation

Parents

Loading