Shardy is an MLIR-based partitioning system. It is meant to be dialect agnostic and provide extensive user control and debuggability features. It includes an axis-based sharding representation, a set of compiler APIs, functionality for sharding propagation, and plans for an SPMD partitioner.
For more information see the openxla/shardy repository.
Communication channels
- Issues and feature requests can be filed in the openxla/shardy repository.
- Questions regarding Shardy can be asked on the OpenXLA Discord.
Status
Shardy is a work in progress. The core dialect and C bindings are fully functional. The python bindings are under development.