OpenTPU 项目使用教程

1. 项目的目录结构及介绍

OpenTPU 项目的目录结构如下：

OpenTPU/
├── gen_one_hot.py
├── simple_nn.py
├── sim.py
├── checker.py
├── config.py
├── README.md
├── LICENSE
└── ...

• gen_one_hot.py: 生成随机训练数据和标签的脚本。
• simple_nn.py: 训练一个简单的两层神经网络的脚本。
• sim.py: 硬件模拟器脚本。
• checker.py: 结果验证脚本。
• config.py: 项目配置文件。
• README.md: 项目介绍和使用说明。
• LICENSE: 项目许可证。

2. 项目的启动文件介绍

项目的启动文件主要包括以下几个脚本：

• gen_one_hot.py: 用于生成训练数据和标签。 示例用法：

  python gen_one_hot.py --path simple_train --shape 8 8 --range -5 5
  python gen_one_hot.py --path simple_train_label --shape 8 1 --range 0 2
  

• simple_nn.py: 用于训练一个简单的两层神经网络。 示例用法：

  python simple_nn.py --train_data simple_train.npy --label_data simple_train_label.npy
  

• sim.py: 硬件模拟器脚本，用于模拟硬件行为。 示例用法：

  python sim.py --input boston_input.npy --weights boston_weights.npy
  

• checker.py: 用于验证硬件模拟器和应用程序的结果。 示例用法：

  python checker.py --sim_output sim_results.npy --app_output app_results.npy
  

3. 项目的配置文件介绍

项目的配置文件是 config.py，其中包含以下配置项：

• Unified Buffer size: 统一缓冲区的大小。
• Accumulator Buffer size: 累加器缓冲区的大小。
• MM Array size: 矩阵乘法阵列的大小，必须是方形。

示例配置：

# config.py
UNIFIED_BUFFER_SIZE = 1024
ACCUMULATOR_BUFFER_SIZE = 512
MM_ARRAY_SIZE = 256

通过修改 config.py 文件中的配置项，可以调整硬件模拟器的参数。