OpenTPU

UCSB ArchLab OpenTPU 项目

OpenTPU 是 UC Santa Barbara ArchLab 重新实现 Google 的 Tensor 处理单元 (TPU) 的开源项目。 TPU 是 Google 为加速 神经网络 计算的推理阶段而定制的 ASIC。 我们的设计基于 Google 发表在 ISCA2017 的论文《数据中心内张量处理单元的性能分析》（https://arxiv.org/abs/1704.04760）中的细节。然而，对于 TPU 尚未发布任何正式规范、接口或 ISA。

OpenTPU 由 PyRTL (http://ucsbarchlab.github.io/PyRTL/) 提供支持。

要求

• Python 3
• PyRTL 版本 >= 0.8.5
• numpy PyRTL 和 numpy 都可以使用 pip 安装;例如,pip install pyrtl。

如何运行

要在硬件和功能模拟器中运行简单的矩阵乘法测试，请确保在 config.py 中将 MATSIZE 设置为 8，然后运行 ``` python3 assembler.py simplemult.a python3 runtpu.py simplemult.out simplemult_hostmem.npy simplemult_weights.npy python3 sim.py simplemult.out simplemult_hostmem.npy simplemult_weights.npy

请注意,硬件矩阵乘法单元的大小可以参数化 - 请仔细检查 `config.py` 以确保 MATSIZE 符合您的预期。