使用方式

  1. 编译项目
    进入示例目录并执行编译脚本:

    cd examples/allgather_matmul
    bash scripts/build.sh
    
  2. 运行AllGather-MatMul示例程序
    在示例目录下执行运行脚本:

    bash scripts/run.sh <device_list>
    
    • 参数说明

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