CANN/shmem RDMA性能测试示例

环境要求

【免费下载链接】shmemCANN SHMEM 是面向昇腾平台的多机多卡内存通信库，基于OpenSHMEM 标准协议，实现跨设备的高效内存访问与数据同步。项目地址: https://gitcode.com/cann/shmem

同rdma_demo中的环境要求。

使用方式

1.在shmem/目录编译:

bash scripts/build.sh -enable_rdma -examples

2.在shmem/目录运行:

export PROJECT_ROOT=<shmem-root-directory> export LD_LIBRARY_PATH=${PROJECT_ROOT}/build/lib:$LD_LIBRARY_PATH export SHMEM_UID_SESSION_ID=127.0.0.1:8899 ./build/bin/rdma_perftest 2 0 tcp://127.0.0.1:8899 2 0 0 highlevel_put_pingpong_latency 64 # pe 0 ./build/bin/rdma_perftest 2 1 tcp://127.0.0.1:8899 2 1 1 highlevel_put_pingpong_latency 64 # pe 1

3.命令行参数说明 ./rdma_perftest <n_pes> <pe_id> <g_npus> <f_pe> <f_npu> <test_type> <msg_len>

• n_pes: 全局pe数量，只支持2个pe。
• pe_id: 当前进程的pe号。
• ipport: SHMEM初始化需要的IP及端口号，格式为tcp:// :<端口号>。如果执行跨机测试，需要将IP设为pe0所在Host的IP。
• g_npus: 当前卡上启动的NPU数量。
• f_pe: 当前卡上使用的第一个pe号。
• f_npu: 当前卡上使用的第一个NPU卡号。
• test_type: 测试类型。
  • highlevel_put_pingpong_latency：测试Put高阶接口的pingpong时延。
  • postsend_cost: 测试postsend接口耗时。
  • highlevel_put_bw: 测试Put高阶接口的带宽。
  • rdma_mte_bw: 测试并行下发MTE和RDMA时的带宽。
• msg_len: 测试传输的数据量大小，单位为字节（Byte）。

【免费下载链接】shmemCANN SHMEM 是面向昇腾平台的多机多卡内存通信库，基于OpenSHMEM 标准协议，实现跨设备的高效内存访问与数据同步。项目地址: https://gitcode.com/cann/shmem