jax
c6525bc5 - [Mosaic GPU][NFC] Fix documentation of `WGMMA_LAYOUT`.

Commit

1 year ago

[Mosaic GPU][NFC] Fix documentation of `WGMMA_LAYOUT`. `TiledLayout` has no notion of partitioning over warpgroups, and each warp holds `16 x 8` elements. PiperOrigin-RevId: 739942481

References

#27365 - [Mosaic GPU][NFC] Fix documentation of `WGMMA_LAYOUT`.

Author

bchetioui

Committer

Google-ML-Automation

Parents

01904363

jax c6525bc5 - [Mosaic GPU][NFC] Fix documentation of `WGMMA_LAYOUT`.

jax
c6525bc5 - [Mosaic GPU][NFC] Fix documentation of `WGMMA_LAYOUT`.