Chip
函数功能
芯片抽象，在with语句中实例化并使用来明确针对某一昇腾AI处理器类型进行建模。
函数原型
class Chip(name, debug_mode=False)
参数说明
参数名 |
输入类型 |
说明 |
|---|---|---|
name |
string |
芯片名称。 目前大部分数据基于Atlas A2训练系列产品/Atlas 800I A2推理产品采集，使用npu-smi info可以查看当前设备昇腾AI处理器类型。 |
debug_mode |
bool |
是否启用调试模式，默认为False。
开启debug模式后可查看未正确运行的指令。 |
成员
成员名称 |
描述 |
|---|---|
chip.enable_trace() |
使能算子模拟流水图的功能，生成流水图文件trace.json。 |
chip.enable_metrics() |
使能单指令及分PIPE的流水信息，生成指令统计（Instruction_statistic.csv）、搬运流水统计（Pipe_statistic.csv）文件和指令占比饼图（instruction_cycle_consumption.html）。 |
约束说明
需在with语句下将该类初始化。
使用示例
from mskpp import Chip
# 使用npu-smi info可以查看当前设备昇腾芯片类型
with Chip("Ascendxxx") as chip:
chip.enable_trace() # 调用该函数即可使能算子模拟流水图的功能，生成流水图文件
chip.enable_metrics() # 调用该函数即可使能单指令及分PIPE的流水信息，生成搬运流水统计、指令信息统计和指令占比饼图
父主题： mskpp对外接口使用说明