MLIR中对稀疏张量计算的编译器支持
摘要：在科学、工程、机器学习和数据分析问题中出现了稀疏张量。操作这些张量的程序可以利用稀疏性来减少存储需求和计算时间。然而，手动开发和维护稀疏软件是一项复杂且容易出错的任务。因此，我们提出将稀疏性视为张量的属性，而不是繁琐的实现任务，并让稀疏编译器从稀疏不可知的计算定义自动生成稀疏代码。本文讨论了将这个想法整合到MLIR中。
作者：Aart J.C. Bik, Penporn Koanantakool, Tatiana Shpeisman, Nicolas Vasilache, Bixia Zheng, Fredrik Kjolstad
论文ID：2202.04305
分类：Programming Languages
分类简称：cs.PL
提交时间：2022-09-20