概述
本章节包含采集profiling相关的自定义接口,提供性能优化所需要的数据。
| API名称 | 说明 | 
|---|---|
| 提供PyTorch训练过程中的性能数据采集功能。 | |
| 提供PyTorch训练过程中的性能数据采集功能。 | |
| 事件采集列表,枚举类。用于赋值给torch_npu.profiler.profile的activities参数。 | |
| 将采集到的性能数据导出为TensorBoard工具支持的格式。作为torch_npu.profiler.profile on_trace_ready参数的执行操作。 | |
| 设置不同step的行为。用于构造torch_npu.profiler.profile的schedule参数。 | |
| Profiler状态,Enum类型。 | |
| 性能数据采集扩展参数。用于构造torch_npu.profiler.profile的experimental_config参数。 | |
| 设置导出的性能数据结果文件格式,作为 _ExperimentalConfig类的export_type参数。 | |
| 采集等级,作为 _ExperimentalConfig类的profiler_level参数。 | |
| AI Core的性能指标采集项,作为 _ExperimentalConfig类的aic_metrics参数。 | |
| 查询当前支持采集的activities参数的CPU、NPU事件。 | |
| 查询当前支持的torch_npu.profiler.ProfilerLevel级别。 | |
| 查询当前支持的torch_npu.profiler. AiCMetrics的AI Core性能指标采集项。 | |
| 查询当前支持的torch_npu.profiler.ExportType的性能数据结果文件类型。 | |
| 初始化dynamic_profile动态采集。 | |
| dynamic_profile动态采集划分step。 | |
| 触发一次dynamic_profile动态采集。 | |
| Ascend PyTorch Profiler性能数据离线解析。 |