PrIM (Processing-In-Memory) 基准测试项目使用教程

1. 项目目录结构及介绍

PrIM (Processing-In-Memory) 基准测试项目是一个用于评估和分析处理内存（PIM）架构的基准测试套件。以下是项目的目录结构及其介绍：

prim-benchmarks/
├── LICENSE
├── README.md
├── run_strong_full.py
├── run_strong_rank.py
├── run_weak.py
├── BFS/
│   ├── baselines/
│   │   ├── cpu/
│   │   └── gpu/
│   ├── data/
│   ├── dpu/
│   ├── host/
│   ├── support/
│   └── Makefile
├── BS/
├── GEMV/
├── HST-L/
├── HST-S/
├── MLP/
├── Microbenchmarks/
│   ├── Arithmetic-Throughput/
│   ├── CPU-DPU/
│   ├── MRAM-Latency/
│   ├── Operational-Intensity/
│   ├── Random-GUPS/
│   ├── STREAM/
│   ├── STRIDED/
│   └── WRAM/
├── NW/
├── RED/
├── SCAN-RSS/
├── SCAN-SSA/
├── SEL/
├── SpMV/
├── TRNS/
├── TS/
├── UNI/
└── VA/

目录结构介绍

• LICENSE: 项目的许可证文件。
• README.md: 项目的介绍和使用说明。
• run_strong_full.py: 用于运行强扩展实验的脚本。
• run_strong_rank.py: 用于运行强扩展实验的脚本。
• run_weak.py: 用于运行弱扩展实验的脚本。
• BFS/: 广度优先搜索（BFS）基准测试目录。
  • baselines/: 包含CPU和GPU的基准实现。
  • data/: 包含测试数据。
  • dpu/: 包含DPU（DRAM Processing Unit）的实现。
  • host/: 包含主机端的实现。
  • support/: 包含支持文件。
  • Makefile: 用于编译BFS基准测试的Makefile。
• BS/, GEMV/, HST-L/, HST-S/, MLP/, NW/, RED/, SCAN-RSS/, SCAN-SSA/, SEL/, SpMV/, TRNS/, TS/, UNI/, VA/: 其他基准测试目录，结构与BFS类似。
• Microbenchmarks/: 包含微基准测试目录，用于评估架构的各个方面。

2. 项目启动文件介绍

项目中有多个启动文件，用于运行不同类型的实验。以下是主要的启动文件及其功能介绍：

• run_weak.py: 用于运行弱扩展实验。弱扩展实验是指在增加DPU数量的同时，保持每个DPU的工作负载不变。
• run_strong_rank.py: 用于运行强扩展实验。强扩展实验是指在增加DPU数量的同时，增加总的工作负载。
• run_strong_full.py: 用于运行强扩展实验，但使用多个rank（多个DPU组）。

使用示例

# 运行弱扩展实验
python3 run_weak.py BFS

# 运行强扩展实验
python3 run_strong_rank.py BFS

# 运行强扩展实验（多rank）
python3 run_strong_full.py BFS

3. 项目配置文件介绍

项目中没有明确的配置文件，但可以通过修改启动脚本中的参数来配置实验。例如，在run_weak.py中，可以通过修改rootdir变量来指定项目的根目录。

rootdir = "/"  # 修改为项目的实际路径

此外，每个基准测试目录中的Makefile也包含一些配置参数，例如NR_DPUS和NR_TASKLETS，用于指定DPU的数量和每个DPU的软件线程数。

示例

cd BFS
# 编译BFS基准测试，使用32个DPU和16个软件线程
NR_DPUS=32 NR_TASKLETS=16 make all

通过这些配置，可以灵活地调整实验的参数，以适应不同的测试需求。