昇腾社区首页
中文
注册
开发者
下载

torch_npu.profiler.ProfilerLevel

产品支持情况

[object Object]undefined

功能说明

采集等级,Enum类型。用于作为_ExperimentalConfig类的profiler_level参数。

函数原型

[object Object]

参数说明

  • torch_npu.profiler.ProfilerLevel.Level_none:可选参数,不采集所有Level层级控制的数据,即关闭profiler_level。

  • torch_npu.profiler.ProfilerLevel.Level0:可选参数,采集上层应用数据、底层NPU数据以及NPU上执行的算子信息。配置该参数时,仅采集部分数据,其中部分算子信息不采集,详细情况请参见《CANN 性能调优工具用户指南》中的“[object Object]op_summary(算子详细信息)[object Object]”章节。

  • torch_npu.profiler.ProfilerLevel.Level1:可选参数,在Level0的基础上多采集CANN层AscendCL数据和NPU上执行的AI Core性能指标信息、开启aic_metrics=torch_npu.profiler.AiCMetrics.PipeUtilization以及HCCL的communication.json和communication_matrix.json文件。

  • torch_npu.profiler.ProfilerLevel.Level2:可选参数,在Level1的基础上多采集CANN层Runtime数据以及AI CPU(data_preprocess.csv文件)。

默认值为torch_npu.profiler.ProfilerLevel.Level0。

返回值说明

调用示例

以下是关键步骤的代码示例,不可直接拷贝编译运行,仅供参考。

[object Object]