torch_npu.profiler.ProfilerActivity
产品支持情况
| 产品 | 是否支持 |
|---|---|
| [object Object]Atlas A3 训练系列产品[object Object] | √ |
| [object Object]Atlas A2 训练系列产品[object Object] | √ |
| [object Object]Atlas 训练系列产品[object Object] | √ |
功能说明
事件采集列表，Enum类型。用于赋值给torch_npu.profiler.profile的activities参数。
函数原型
[object Object]
参数说明
- torch_npu.profiler.ProfilerActivity.CPU：可选参数，框架侧数据采集的开关。
- torch_npu.profiler.ProfilerActivity.NPU：可选参数，CANN软件栈及NPU数据采集的开关。
默认情况下两个开关同时开启。
返回值说明
无
调用示例
以下是关键步骤的代码示例，不可直接拷贝编译运行，仅供参考。
[object Object]