Shardy

Shardy 是基于 MLIR 的分区系统。它不依赖于方言,并提供丰富的用户控制和调试功能。它包括基于轴的分片表示法、一组编译器 API、分片传播功能以及 SPMD 分区器的方案。

如需了解详情,请参阅 openxla/shardy 代码库。

通信通道

状态

Shardy 仍在开发阶段。核心方言和 C 绑定完全正常运行。Python 绑定正在开发中。