jax
b19ad5b3 - [Mosaic GPU] Add support for non-128B swizzles in WGMMA

Commit

1 year ago

[Mosaic GPU] Add support for non-128B swizzles in WGMMA PiperOrigin-RevId: 647667550

References

Author

apaszke

Committer

a-googler

Parents