CANN/shmem RDMA性能测试示例

张开发
2026/5/9 21:03:55 15 分钟阅读

分享文章

CANN/shmem RDMA性能测试示例
环境要求【免费下载链接】shmemCANN SHMEM 是面向昇腾平台的多机多卡内存通信库基于OpenSHMEM 标准协议实现跨设备的高效内存访问与数据同步。项目地址: https://gitcode.com/cann/shmem同rdma_demo中的环境要求。使用方式1.在shmem/目录编译:bash scripts/build.sh -enable_rdma -examples2.在shmem/目录运行:export PROJECT_ROOTshmem-root-directory export LD_LIBRARY_PATH${PROJECT_ROOT}/build/lib:$LD_LIBRARY_PATH export SHMEM_UID_SESSION_ID127.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 13.命令行参数说明 ./rdma_perftest n_pes pe_id g_npus f_pe f_npu test_type msg_lenn_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创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考

更多文章