torch_npu.npu.profile

API接口

torch_npu.npu.profile(profiler_result_path, use_e2e_profiler, config)

功能描述

通过profiling获取性能数据文件。

参数说明

• profiler_result_path：profiling结果保存路径，默认为当前路径。
• use_e2e_profiler：是否开启E2E profiling功能，默认为False，代表仅开启CANN profiling功能，采集CANN层面的数据。
• config：torch.npu. profileConfig类型，指定Profiling配置数据。