环境要求

同rdma_demo中的环境要求。

使用方式

1.在shmem/目录编译:

bash scripts/build.sh -enable_rdma -examples

2.直接在examples/rdma_handlewait_test/use_handlewait目录下执行bash run.sh；或者在shmem/目录运行:

• 单机2卡执行命令

  export PROJECT_ROOT=<shmem-root-directory>
  export LD_LIBRARY_PATH=${PROJECT_ROOT}/build/lib:$LD_LIBRARY_PATH
  ./build/bin/use_handlewait 2 0 tcp://127.0.0.1:8765 2 0 0 & # PE 0
  ./build/bin/use_handlewait 2 1 tcp://127.0.0.1:8765 2 0 0 & # PE 1
  

  注：<shmem-root-directory>为SHMEM项目的根目录。

• 跨机2卡执行命令

  假设机器A的ip为ip1，机器B的ip为ip2。 在机器A执行如下命令：

  export PROJECT_ROOT=<shmem-root-directory>
  export LD_LIBRARY_PATH=${PROJECT_ROOT}/build/lib:$LD_LIBRARY_PATH
  ./build/bin/use_handlewait 2 0 tcp://ip1:8765 1 0 0 # PE 0
  

  同时，在机器B执行如下命令：

  export PROJECT_ROOT=<shmem-root-directory>
  export LD_LIBRARY_PATH=${PROJECT_ROOT}/build/lib:$LD_LIBRARY_PATH
  ./build/bin/use_handlewait 2 1 tcp://ip1:8765 1 1 0 # PE 1
  

  注：<shmem-root-directory>为SHMEM项目的根目录。

3.命令行参数说明

    ./use_handlewait <n_pes> <pe_id> <ipport> <g_npus> <f_pe> <f_npu>

• n_pes: 全局PE数量。
• pe_id: 当前进程的PE号。
• ipport: SHMEM初始化需要的IP及端口号，格式为tcp://:<端口号>。如果执行跨机测试，需要将IP设为PE0所在Host的IP。
• g_npus: 当前机器上启动的NPU卡的数量。
• f_pe: 当前机器上使用的第一个PE号。
• f_npu: 当前机器执行本样例使用的第一张NPU卡的卡号