jax
cbae2539 - [mgpu:pallas] Typo in `UnswizzleRef.untransform_reshape()` check.

Commit

1 year ago

[mgpu:pallas] Typo in `UnswizzleRef.untransform_reshape()` check. PiperOrigin-RevId: 743920665

Author

cperivol

Committer

Google-ML-Automation

Parents