Tensor 代数编译器 (taco) 可在 CPU 和 GPU 上计算稀疏张量表达式
Tensor 代数编译器 (taco) 是一个 C++ 库,用于在稀疏和密集张量上计算张量代数表达式。 它使用新的编译器技术,使性能与广泛使用的稀疏张量代数和稀疏线性代数库中手动优化的内核相媲美。
更多关于 taco 的信息请参阅 tensor-compiler.org、论文 The Tensor Algebra Compiler 或 此演讲。要了解 taco 近期发展的更多信息,请参阅有关 优化 和 格式 的技术报告。您还可以订阅我们发布公告、RFC 和 API 变更通知的电子邮件列表,或订阅 taco-discuss 电子邮件列表,进行开放式讨论和提问。
构建 taco 需要 gcc 5.0 或更高版本,或 clang 3.9 或更高版本。您可以通过在运行 cmake 之前设置 CC 和 CXX 环境变量来使用特定的编译器或版本。例如:
这需要 Python 3.x 和一些开发库。 还需要安装 NumPy 和 SciPy。 对于 Debian/Ubuntu, 需要以下包: `python3 libpython3-dev python3-distutils python3-numpy py
暂无开放 Issues,或尚未同步最近议题。