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语句下将该类初始化。
使用示例
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对外接口使用说明
