jax
b7f91bfd - [mgpu] Add support for CTA barriers in Mosaic GPU.

Commit

30 days ago

[mgpu] Add support for CTA barriers in Mosaic GPU. PiperOrigin-RevId: 999138269

References

#41471 - [mgpu] Add support for CTA barriers in Mosaic GPU.

Author

chr1sj0nes

chr1sj0nes

Committer

Google-ML-Automation

Google-ML-Automation

Parents

Loading