带宽测试
测试项功能
带宽测试主要用于测试总线带宽、内存带宽和时延。
注意事项
- 为了避免频繁输出日志影响测试结果,测试前确认Host和Device的日志级别设置为ERROR,确认及设置方法如下:
- d2d带宽测试结果通过数据读写总量/消耗时间获取。因为d2d带宽测试和实际训练或推理一样,存在缓存、预取等内在优化,由此计算得到的带宽结果有可能超过标称带宽。
- Atlas 200I SoC A1核心板的测试数据流向为h2d和d2h方式时,由于架构的特殊性,测试结果是从CPU直接拷贝获取的,结果和其他产品形态不一致,属于正常现象。
- 为确保带宽测试结果达到最佳,用户需在裸机上进行测试。
测试项参数查询
用户可任选以下指令之一查看带宽测试命令的可用参数。
ascend-dmi --bw -h
ascend-dmi --bw --help
命令各参数解释如表1所示。
参数 |
说明 |
是否必填 |
---|---|---|
[-bw, --bw, --bandwidth] |
使用该参数测试芯片的带宽。支持-bw,但建议使用--bw或--bandwidth。 |
是 |
[-t, --type] |
指测试数据流向的分类。
当使用带宽和时延测试功能时,测试的数据流可以分为以下方向,若不填写数据流方向则默认返回h2d、d2h、d2d三个方向的带宽和时延信息。
|
否 |
[-s, --size] |
指传输数据大小并指定测试结果显示方式,传输数据的取值范围为1Byte~512M,单位为字节。 显示方式分为定长模式和步长模式。若“-s”参数缺省则为步长模式,输出传输数据的带宽测试结果,传输数据的范围为2Byte~32M。若“-s”参数不缺省则为定长模式,“-s”参数后必须填写数值指定传输数据的大小,不填写属于错误写法。 除此之外,当使用p2p模式,若“-s”参数缺省,默认传输数据值为128M。若“-s”参数不缺省,“-s”参数后必须填写数值指定传输数据的大小,不填写属于错误写法。 |
否 |
[-et, --et, --execute-times] |
指迭代次数,即内存拷贝次数。取值范围为[1, 1000],若不填写拷贝次数则默认为5。 |
否 |
[-d, --device] |
指定需要测试带宽的Device ID,Device ID是指昇腾芯片的ID,用户可以执行ascend-dmi --info命令,在显示界面表格中的Chip参数处获得芯片数量。比如一个Atlas 300I 推理卡配置4个昇腾芯片,则Device ID的取值范围为[0,3]。若不填写Device ID则默认返回Device 0带宽信息。 |
否 |
[-ds, --ds, --device-src] |
指定p2p测试的源头Device的ID号。必须与[-dd, --dd, --device-dst]参数成对指定。 |
否 |
[-dd, --dd, --device-dst] |
指定p2p测试的目标Device的ID号。必须与[-ds, --ds, --device-src]参数成对指定。 |
否 |
[-fmt, --fmt, --format] |
指定输出格式,可以为normal或json。若未指定则默认为normal。 |
否 |

- --ds与--dd参数需要配合使用,单独使用是错误用法,使用时参数后的数值不能相同。
- ascend-dmi --bw后使用-t,-s等多个二级参数时,可任意指定这些参数的排列顺序,不影响命令结果输出。例如:ascend-dmi --bw -t h2d -d 0 --et 100和ascend-dmi --bw -t h2d --et 100 -d 0输出结果相同。
- 由于NPU的工作模式会影响p2p带宽测试的计算方式,若出现p2p带宽测试结果与标称带宽相差过大,建议用户使用SMP模式。操作如下:用户登录IBMC,执行如下命令设置为SMP模式,其中1为SMP,0为AMP。
使用实例
使用实例中命令的回显在推理服务器与训练服务器类似,截图取自推理服务器,此处以Ascend 310芯片为例。需要注意,p2p测试命令截图取自训练服务器,因为p2p测试仅支持Ascend 910系列芯片设备,即训练服务器。
- 以不带参数为例(不带参数则默认查询在Device 0,以h2d、d2h、d2d三个数据流向和步长模式显示的带宽时延信息)。
- 以测试数据从Host侧传输到Device 0,迭代100次的带宽与时延为例。
- 测试指定源头Device到目标Device的传输速率和时延。
- 以测试数据从源头Device 1传输到目标Device 2的p2p测试为例。
ascend-dmi --bw -t p2p --ds 1 --dd 2
若返回如图3所示信息,表示工具运行正常,图中参数介绍如表2所示。
表2 显示界面参数介绍 参数
说明
Host to Device Test
带宽数据流方向。有以下显示可能:- Host to Device Test
- Device to Host Test
- Device to Device Test
- Unidirectional Peer to Peer Test
- Bidirectional Peer to Peer Test
Device X : Ascend XXX
Device X为当前测试的源设备ID,Ascend XXX为芯片类型。
ID
Device ID
Size(Bytes)
传输数据大小
Execute Times
迭代次数
Bandwidth(MB/s)
芯片的带宽
Elapsed Time(us)
执行时长
- 不指定源头Device和目标Device的p2p样例。
若返回如图4所示信息,表示工具运行正常。
- 以测试数据从源头Device 1传输到目标Device 2的p2p测试为例。