论文标题
妈妈：MLIR的矩阵操作
MOM: Matrix Operations in MLIR
论文作者
论文摘要
对密度线性代数计算的代码生成器中的现代研究表明，能够通过比较域专家进行比较并经常超过最先进的实现之一的性能。但是，基础架构通常是通过隔离而开发的，这使得在逻辑上可组合的系统的互连即使不是不可能。在本文中，我们建议利用MLIR作为统一的编译器基础架构来优化密集的线性代数操作。我们提出了一种新的MLIR方言，用于表达线性代数计算，包括矩阵属性，以实现高级算法转换。在MLIR中，该新方言的集成通过转换为框架已经提供的现有低级方言来实现矩阵计算的端到端汇编。
Modern research in code generators for dense linear algebra computations has shown the ability to produce optimized code with a performance which compares and often exceeds the one of state-of-the-art implementations by domain experts. However, the underlying infrastructure is often developed in isolation making the interconnection of logically combinable systems complicated if not impossible. In this paper, we propose to leverage MLIR as a unifying compiler infrastructure for the optimization of dense linear algebra operations. We propose a new MLIR dialect for expressing linear algebraic computations including matrix properties to enable high-level algorithmic transformations. The integration of this new dialect in MLIR enables end-to-end compilation of matrix computations via conversion to existing lower-level dialects already provided by the framework.
