解析并提取dump信息中的堆栈信息及数据统计信息。
parse(pkl_file, moudule_name_prefix)
参数名 |
说明 |
是否必选 |
|---|---|---|
pkl_file |
指定dump数据文件中的pkl文件名。参数示例："./npu_dump/ptdbg_dump_v4.0/step0/rank0/dump.pkl"。 |
是 |
moudule_name_prefix |
指定待提取的API接口前缀。参数示例："Torch_norm_1_forward"。 |
是 |
创建堆栈信息及数据统计信息提取脚本，例如parse.py，拷贝如下代码，具体参数请根据实际环境修改。
from ptdbg_ascend import *
parse("./npu_dump/ptdbg_dump_v4.0/step0/rank0/dump.pkl", "Torch_batch_normal_1_forward")