SuperAPI AI 工具集

ascend-dmi

部署与运维

查看项目

更新于 10-15 17:19

技能介绍

用途:对华为昇腾 NPU 做硬件层面的管理、测试与诊断,包括查看 NPU 卡状态、温度、利用率,测试 h2d/d2h/d2d/p2p 内存带宽,运行 TFLOPS/TOPS 算力与功耗基准,执行健康检查、性能规格检查、压力测试、眼图测试和 NPU 恢复。

主要能力与命令模板:版本 ascend-dmi -v;设备状态 ascend-dmi -i [-b|--dt];兼容性检查 ascend-dmi -c;眼图测试 ascend-dmi --sq -t pcie|hbm|roce|all [-d ID];带宽测试 ascend-dmi --bw -t h2d|d2h|d2d|p2p [-d ID] [-s SIZE] [-q];算力测试 ascend-dmi -f -t fp16|fp32|int8|bf16|hf32 [-d ID] [--all] [-q];功耗测试 ascend-dmi -p [-pt edp|tdp] [--dur SEC] [-q];健康检查 ascend-dmi --dg --se healthCheck [-d ID] [-q];性能规格 ascend-dmi --dg --se performanceCheck [-d ID] [-q];压力测试 ascend-dmi --dg --se stressTest [-d ID] [-q] 或 -s -i ITEM;NPU 恢复 ascend-dmi -r -d ID [-q]。

使用方式:按用途选择路径(信息查询与眼图测试为快速路径,性能测试与健康检查为标准路径,压力测试与恢复为高风险路径)。先检查 ascend-dmi:which ascend-dmi && ascend-dmi -v,不可用时执行 source /usr/local/Ascend/toolbox/set_env.sh 后重试。性能、诊断、压力类操作还需验证 CANN:python3 -c "import acl; print(acl.get_soc_name())",输出芯片型号即可用;失败需由用户提供 CANN 路径或经同意后自动搜索并每次验证。远程执行须先建立安全连接,不可直接 ssh。执行前用 npu-smi info 查看设备,并逐项确认测试模式、设备 ID、数据大小、数据类型、时长等参数。中风险操作会独占 NPU、中断训练推理,需确认后加 -q;高风险操作需告知风险、用 fuser -v /dev/davinci* 检查占用并二次确认。输出解读关注 Bandwidth(GB/s)、TFLOPS/TOPS(--all 为整机总和)、Power(W) 与 Temperature(°C)(应低于 90°C),诊断结果为 PASS/FAIL/SKIP/WARN。

适用场景:昇腾 NPU 硬件状态巡检、性能与功耗基准、故障排查与压力验证。限制:D2D 在 A2/A3 上不支持 -s;aicpu 压力项须单独执行;数据类型支持随产品不同;具体产品规格、价格与部署细节暂未提供。