当使用Ascend PyTorch Profiler接口采集的性能数据较大时,若在当前环境直接使用on_trace_ready接口进行自动解析,则可能导致资源占用过大出现卡顿,那么可以取消on_trace_ready接口,并通过环境变量ASCEND_WORK_PATH设置落盘目录(例如:export ASCEND_WORK_PATH=xx/xx),在采集完成性能数据后,使用如下方式进行离线解析:
1 2 3 4 | from torch_npu.profiler.profiler import analyse if __name__ == "__main__": analyse(profiler_path="./result_data", max_process_number=max_process_number, export_type=export_type) |
参数 |
描述 |
可选/必选 |
---|---|---|
profiler_path |
PyTorch性能数据路径。路径格式仅支持由字母、数字和下划线组成的字符串,不支持软链接。指定的目录下保存PyTorch性能数据目录{worker_name}_{时间戳}_ascend_pt。 |
必选 |
max_process_number |
离线解析最大进程数。取值范围为1~CPU核数,默认为CPU核数的一半。若设置超过该环境的CPU核数,则自动取CPU核数;若设置为非法值,则取默认值CPU核数的一半。 |
可选 |
export_type |
设置导出的性能数据结果文件格式,List类型。可取值:
设置无效值或未配置时,则读取profiler_info.json中的export_type字段,确定导出格式。两个参数可同时配置,表示同时导出timeline、summary和db文件。 解析结果数据请参见Ascend PyTorch Profiler接口采集数据。 |
可选 |
python3 {file_name}.py
性能数据结果文件详细介绍请参见数据目录说明,性能数据分析请参见《MindStudio Insight 用户指南》将解析后的性能数据文件进行可视化展示和分析。