dump文件目录结构
样例dump数据结果目录如下:
|── dump_path | └─ptdbg_dump_{version} | |── rank0 | | |── myDump | | | |── Tensor_permute_1_forward.npy | | | ... | | | └── Function_linear_5_backward_output.npy | | └── myDump.pkl | |── rank1 | | |── myDump | | | └── ... | | └── myDump.pkl | |── rank2 | | |── myDump | | | └── ... | | └── myDump.pkl | |── ... | | | └── rank7
- 在精度比对dump场景下,假设配置的dump文件名为npu_dump.pkl,此时dump的结果为两部分:
- 文件npu_dump.pkl 中包含dump数据的api名称、dtype、 shape、统计信息:max, min, mean。
- 文件夹npu_dump_{timestamp},文件夹下为numpy格式的dump数据.numpy文件保存的前缀和PyTorch对应关系如下:
前缀
Torch模块
Tensor
torch.Tensor
Torch
torch
Functional
torch.nn.functional
NPU
NPU亲和算子
VF
torch._VF
- 当dump模式配置为"api_stack"时,dump结果的文件名会添加api_stack前缀,dump结果如下:
- api_stack_npu_dump.pkl。
- api_stack_npu_dump_timestamp目录。
- 溢出检测dump场景下,若检测到API溢出,此时不需要配置set_dump_path,dump执行时会在当前目录自动生成ptdbg_dump_{version}目录,dump结果如下:
- Overflow_info_{timestamp}.pkl。
- Overflow_info_{timestamp}目录。
父主题: 精度比对工具说明