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

torch_npu.profiler.ProfilerActivity

API接口

torch_npu.profiler.ProfilerActivity

功能描述

事件采集列表,枚举类。用于赋值给torch_npu.profiler.profile的activities参数。

参数说明

  • torch_npu.profiler.ProfilerActivity.CPU:框架侧数据采集的开关。
  • torch_npu.profiler.ProfilerActivity.NPU:CANN软件栈及NPU数据采集的开关。