jax
8e86bede - [Mosaic GPU] Allow multiple gmem indexers on copies.

Commit

1 year ago

[Mosaic GPU] Allow multiple gmem indexers on copies. This is implemented by merging multiple indexers into one. PiperOrigin-RevId: 714150733

References

Author

justinjfu

Committer

Google-ML-Automation

Parents