使用方式

  1. 编译项目
    shmem/ 根目录下执行编译脚本:

    bash scripts/build.sh -examples
    
  2. 运行MatMul-AllReduce示例程序
    进入示例目录并执行运行脚本:

    cd examples/matmul_allreduce
    bash scripts/run.sh [device_list]
    
    • 参数说明

      • device_list:指定用于运行的设备(NPU)编号列表,以逗号分隔。
      • 示例:使用第6和第7个NPU设备运行2卡MatMul-AllReduce示例:
        bash scripts/run.sh 6,7
        
    • 配置计算规模
      矩阵形状参数(M、K、N)可在配置文件 scripts/test_shapes.csv 中进行设置。
      修改该文件以定义测试用例的输入维度。