NVlabs/timeloop

解决的问题

Timeloop 提供了一种对各种加速器架构上的密集和稀疏张量代数工作负载进行建模、映射和代码生成的方法。它帮助设计者避免为不同硬件设计进行性能和能耗预测的手动工作。

工作原理

该系统由两个模块化组件组成：一个用于模拟架构设计以预测性能和能耗的分析模型，以及一个搜索将张量代数问题映射到特定架构的最优方式的映射器。

适用人群

专为从事加速器架构设计和张量代数优化的研究人员和工程师设计。

主要亮点

• 支持密集和稀疏张量代数工作负载。
• 支持压缩稀疏张量代数的随机建模（在 2.0 版本中引入）。
• 支持空间偏移、扁平化映射和非完全因式分解映射（在 3.0 版本中引入）。