昇腾社区首页
中文
注册

torch_npu.profiler._ExperimentalConfig

API接口

torch_npu.profiler._ExperimentalConfig(profiler_level = Constant.LEVEL0, aic_metrics = Constant.AicMetricsNone, l2_cache = False, record_op_args = False)

功能描述

Profiling扩展参数。用于构造torch_npu.profiler.profile的experimental_config参数。

参数说明

  • profiler_level:采集的Level等级,Enum类型。可取值以及含义详见torch_npu.profiler.ProfilerLevel。
  • aic_metrics:AI Core的性能指标采集项,Enum类型,采集的结果数据将在Kernel View呈现。可取值以及含义详见torch_npu.profiler.AiCMetrics
  • l2_cache:控制l2_cache数据采集开关。可取值True或False,默认为False。该采集项在ASCEND_PROFILER_OUTPUT生成l2_cache.csv文件。

示例

experimental_config = torch_npu.profiler._ExperimentalConfig(
    aic_metrics=torch_npu.profiler.AiCMetrics.PipeUtilization, profiler_level=torch_npu.profiler.ProfilerLevel.Level1, 
    l2_cache=False
)