首页
/ PrIM (Processing-In-Memory Benchmarks) 项目下载及安装教程

PrIM (Processing-In-Memory Benchmarks) 项目下载及安装教程

2024-12-06 04:43:46作者:宣聪麟

1. 项目介绍

PrIM (Processing-In-Memory Benchmarks) 是首个针对真实世界处理-内存(PIM)架构的基准测试套件。该项目旨在评估、分析和表征首个公开可用的真实世界 PIM 架构,即 UPMEM PIM 架构。UPMEM PIM 架构将传统的 DRAM 内存阵列与通用顺序核心(称为 DRAM 处理单元,DPU)集成在同一芯片中。PrIM 提供了一组常见的工作负载,用于评估 UPMEM PIM 架构,并可用于编程、架构和系统研究人员,以改进未来 PIM 硬件和软件的多个方面。

2. 项目下载位置

项目代码托管在 GitHub 上,可以通过以下命令进行克隆:

git clone https://github.com/CMU-SAFARI/prim-benchmarks.git

3. 项目安装环境配置

3.1 系统要求

  • 操作系统:Linux(推荐 Ubuntu 20.04 或更高版本)
  • 硬件:支持 UPMEM 模块的服务器
  • 软件:UPMEM SDK

3.2 安装 UPMEM SDK

  1. 下载 UPMEM SDK 安装包。
  2. 解压安装包并运行安装脚本。
tar -xzf upmem-sdk-<version>.tar.gz
cd upmem-sdk-<version>
./install.sh
  1. 配置环境变量。
export UPMEM_HOME=<path_to_upmem_sdk>
export PATH=$PATH:$UPMEM_HOME/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$UPMEM_HOME/lib

3.3 环境配置示例

环境配置示例

4. 项目安装方式

  1. 进入项目目录。
cd prim-benchmarks
  1. 编译项目。
make all
  1. 运行测试。
./bin/host_code -v 0 -f data/loc-gowalla_edges.txt

5. 项目处理脚本

PrIM 项目包含多个处理脚本,用于运行不同类型的实验。以下是一些常用的脚本:

  • run_weak.py:用于运行弱扩展实验。
  • run_strong_rank.py:用于运行强扩展实验(单个 rank)。
  • run_strong_full.py:用于运行强扩展实验(多个 rank)。

5.1 运行弱扩展实验

python3 run_weak.py BFS

5.2 运行强扩展实验(单个 rank)

python3 run_strong_rank.py BFS

5.3 运行强扩展实验(多个 rank)

python3 run_strong_full.py BFS

通过以上步骤,您可以成功下载、安装并运行 PrIM 项目。如果在安装过程中遇到任何问题,请参考项目的 README 文件或联系项目维护者获取帮助。

登录后查看全文
热门项目推荐