jax
824ccd71 - [Shardy] Inline meshes when using shardy and get rid of global meshes from the MLIR body.

Commit

1 year ago

[Shardy] Inline meshes when using shardy and get rid of global meshes from the MLIR body. Also do a couple of cleanups. PiperOrigin-RevId: 685746298

References

Author

yashk2810

Committer

Google-ML-Automation

Parents