jax
11e6cfbc - [Mosaic GPU][NFC] Move the calculation of group strides into _validate_mma

Commit

323 days ago

[Mosaic GPU][NFC] Move the calculation of group strides into _validate_mma This allows us to unify this logic between Hopper and Blackwell. PiperOrigin-RevId: 732862875

References

#26849 - [Mosaic GPU][NFC] Move the calculation of group strides into _validate_mma

#255 - CI: 03/04/25 upstream sync

Author

apaszke

Committer

Google-ML-Automation

Parents

bbadf990

jax 11e6cfbc - [Mosaic GPU][NFC] Move the calculation of group strides into _validate_mma

jax
11e6cfbc - [Mosaic GPU][NFC] Move the calculation of group strides into _validate_mma