Enabling Efficient Sparse Computations Using Linear Algebra Aware Compilers
2 months ago
- #sparse-computations
- #compiler
- #MLIR
- 基于MLIR开发LAPIS编译器框架，用于优化稀疏线性代数运算
- 引入Kokkos方言实现高层代码向不同架构的 lowering，并将MLIR转换为C++ Kokkos代码
- 创建分区方言以支持分布式内存架构，管理稀疏张量分布及通信模式
- 论证MLIR在线性代数级优化的能力，提升各类GPU上的性能表现
- 关键应用领域涵盖稀疏线性代数、图计算内核、TenSQL及子图同构/单态内核
- LAPIS框架全面支持生产力、性能、可移植性及分布式内存执行