API接口
torch_npu.npu.profile(profiler_result_path, use_e2e_profiler, config)
参数说明
- profiler_result_path：profiling结果保存路径，默认为当前路径。
- use_e2e_profiler：是否开启E2E profiling功能，默认为False，代表仅开启CANN profiling功能，采集CANN层面的数据。
- config：torch.npu. profileConfig类型，指定Profiling配置数据。