StableHLO 字节码

MLIR 字节码格式

StableHLO 使用 MLIR 字节码格式进行序列化。

MLIR 字节码格式是一种用于对 MLIR 程序进行编码的序列化格式。从 MLIR RFC 开始,它是出于“二进制格式给表格带来的优势;即序列化速度和大小、mmap 功能,以及更轻松地启用的版本控制等”而构建的。系统使用各种方言的大型测试运行性能、序列化大小和内存测试,以验证格式。

MLIR 字节码并不是专为让 MLIR 保持稳定而构建的,但 MLIR RFC 指出,可以在此格式的基础上提供兼容性保证,我们成功在 StableHLO 上做到了这一点(请参阅 compatibility.md)。