jax
2e6da35e - [Mosaic GPU] Add support for clusters in the matmul example

Commit

1 year ago

[Mosaic GPU] Add support for clusters in the matmul example With the collective async_copy API, the changes are quite minimal! PiperOrigin-RevId: 655937185

References

#22655 - [Mosaic GPU] Add support for clusters in the matmul example

Author

apaszke

Committer

a-googler

Parents

e59303cf

jax 2e6da35e - [Mosaic GPU] Add support for clusters in the matmul example

jax
2e6da35e - [Mosaic GPU] Add support for clusters in the matmul example