jax
93d8a323 - [Mosaic GPU] Add support for s8 matmuls on Blackwell

Commit

233 days ago

[Mosaic GPU] Add support for s8 matmuls on Blackwell This follows our WGMMA implementation. PiperOrigin-RevId: 772896470

References

Author

apaszke

Committer

charleshofer

Parents