使用方式
-
编译项目
在shmem/根目录下执行编译脚本:bash scripts/build.sh -examples -
运行AllGather-MatMul-Padding示例程序
进入示例目录并执行运行脚本:cd examples/allgather_matmul_padding 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中进行设置。
修改该文件以定义测试用例的输入维度。
-