昇腾社区首页
中文
注册

带宽测试

测试项功能

带宽测试主要用于测试总线带宽、内存带宽和时延。

注意事项

  • 为了避免频繁输出日志影响测试结果,测试前确认Host和Device的日志级别设置为ERROR,确认及设置方法如下:
    1. 确认日志级别:
      • Host侧:通过执行echo $GLOBAL_LOG_LEVEL命令查询,如果查询结果为非法值或者空,表示日志级别为缺省级别ERROR,对应数值3。
      • Device侧:请参考CANN 日志参考的“附录>msnpureport工具使用”章节查看全局日志级别、模块日志级别和是否开启Event日志。
    2. 如果日志级别不为ERROR,请参考CANN 日志参考的“设置日志级别”章节设置Host和Device侧的日志级别。
  • d2d带宽测试结果通过数据读写总量/消耗时间获取。因为d2d带宽测试和实际训练或推理一样,存在缓存、预取等内在优化,由此计算得到的带宽结果有可能超过标称带宽。
  • Atlas 200I SoC A1核心板的测试数据流向为h2d和d2h方式时,由于架构的特殊性,测试结果是从CPU直接拷贝获取的,结果和其他产品形态不一致,属于正常现象。
  • 为确保带宽测试结果达到最佳,用户需在裸机上进行测试。

测试项参数查询

用户可任选以下指令之一查看带宽测试命令的可用参数。

ascend-dmi --bw -h

ascend-dmi --bw --help

命令各参数解释如表1所示。

表1 参数说明

参数

说明

是否必填

[-bw, --bw, --bandwidth]

使用该参数测试芯片的带宽。支持-bw,但建议使用--bw或--bandwidth。

[-t, --type]

指测试数据流向的分类。

当使用带宽和时延测试功能时,测试的数据流可以分为以下方向,若不填写数据流方向则默认返回h2d、d2h、d2d三个方向的带宽和时延信息。
  • h2d:指数据从Host侧内存通过PCIe总线搬移到Device侧内存,测试整体带宽及时延。
  • d2h:指数据从Device侧内存通过PCIe总线搬移到Host侧内存,测试整体带宽及时延。
  • d2d:指数据从Device侧内存搬移到同一Device侧内存(主要是用于测试Device侧的内存带宽),测试整体带宽及时延。
  • p2p:测试指定源头Device到目标Device的传输速率和时延,仅支持Ascend 910系列芯片设备。当p2p测试不指定Device时,采用默认方式进行测试,不需要输入size、et、format等参数,输入这些参数也不会影响默认方式的结果。

[-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 100ascend-dmi --bw -t h2d --et 100 -d 0输出结果相同。
  • 由于NPU的工作模式会影响p2p带宽测试的计算方式,若出现p2p带宽测试结果与标称带宽相差过大,建议用户使用SMP模式。操作如下:用户登录IBMC,执行如下命令设置为SMP模式,其中1为SMP,0为AMP。

    ipmcset -d npuworkmode -v 1

使用实例

使用实例中命令的回显在推理服务器与训练服务器类似,截图取自推理服务器,此处以Ascend 310芯片为例。需要注意,p2p测试命令截图取自训练服务器,因为p2p测试仅支持Ascend 910系列芯片设备,即训练服务器。

  • 以不带参数为例(不带参数则默认查询在Device 0,以h2d、d2h、d2d三个数据流向和步长模式显示的带宽时延信息)。

    ascend-dmi --bw

  • 以测试数据从Host侧传输到Device 0,迭代100次的带宽与时延为例。
    • 定长模式。

      ascend-dmi --bw -t h2d -d 0 -s 8388608 --et 100

      若返回如图1所示信息,表示工具运行正常,图中参数介绍如表2所示。

      图1 带宽测试示例(定长模式)
    • 步长模式。

      ascend-dmi --bw -t h2d -d 0 --et 100

      若返回如图2所示信息,表示工具运行正常,图中参数介绍如表2所示。

      图2 带宽测试示例(步长模式)
  • 测试指定源头Device到目标Device的传输速率和时延。
    1. 以测试数据从源头Device 1传输到目标Device 2的p2p测试为例。

      ascend-dmi --bw -t p2p --ds 1 --dd 2

      若返回如图3所示信息,表示工具运行正常,图中参数介绍如表2所示。

      图3 p2p方式带宽测试示例
      表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)

      执行时长

    2. 不指定源头Device和目标Device的p2p样例。

      ascend-dmi --bw -t p2p

      若返回如图4所示信息,表示工具运行正常。

      图4 不指定device的p2p方式带宽测试示例