概述
本章节包含采集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性能数据离线解析。 |