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)。 |
chip.set_cache_hit_ratio(config) |
其中config = {"cache_hit_ratio": 0.6},用于使能手动调整L2Cache命中率,具体介绍请参见支持cache命中率建模。 |
chip.set_prof_summary_path("xxx/PipeUtilization.csv") |
其中PipeUtilization.csv为msprof的结果示例,用于使能pipe信息的理论值与msprof实测值比对。具体介绍请参见支持pipe信息的理论值与msprof实测值比对。 |
约束说明
需在with语句下将该类初始化。
使用示例
1 2 3 4 5 | from mskpp import Chip # 如何查看当前设备昇腾芯片类型请参见以下说明 with Chip("Ascendxxxyy") as chip: # Ascendxxxyy需替换为实际使用的芯片类型 chip.enable_trace() # 调用该函数即可使能算子模拟流水图的功能,生成流水图文件 chip.enable_metrics() # 调用该函数即可使能单指令及分PIPE的流水信息,生成搬运流水统计、指令信息统计和指令占比饼图 |
- 在安装昇腾AI处理器的服务器执行npu-smi info命令进行查询,获取Chip Name信息。实际配置值为AscendChip Name,例如Chip Name取值为xxxyy,实际配置值为Ascendxxxyy。
父主题: msKPP对外接口使用说明