jax
70c90a95 - [Mosaic GPU] Use warpgroup semantics for the ragged dot example kernel.

Commit

1 year ago

[Mosaic GPU] Use warpgroup semantics for the ragged dot example kernel. I verified that the performance doesn't change. PiperOrigin-RevId: 771042037

References

Author

dimitar-asenov

Committer

Google-ML-Automation

Parents