TACO:让稀疏张量计算快几个数量级
Taco: The Tensor Algebra Compiler
TACO 是一个强大的张量代数编译器,专为稀疏线性代数和张量运算而生。它支持从简单的稀疏矩阵向量乘法到复杂的 MTTKRP 高阶稀疏张量运算,兼容 CSR、COO、CSF 等多种存储格式。得益于其创新的编译器技术,TACO 能生成高度优化的计算内核,性能比 MATLAB Tensor Toolbox 高出数个数量级,并与 Eigen、Intel oneMKL、SPLATT 等高性能库媲美。无论是 C++ 还是 Python 开发者,只需几行代码即可实现复杂运算。项目代码托管在 GitHub,提供命令行工具和在线文档,甚至支持直接在浏览器中使用代码生成器。由 Stanford 和 MIT 的研究团队共同维护,TACO 正在重新定义稀疏计算的性能边界。
TACO 采用一种新颖的基于编译器的技术来生成针对您所需运算进行优化的内核,使其性能比 MATLAB Tensor Toolbox 高出数个数量级。