torch_npu.profiler.ProfilerActivity

函数原型

功能说明

事件采集列表，Enum类型。用于赋值给torch_npu.profiler.profile的activities参数。

参数说明

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

默认情况下两个开关同时开启。

支持的型号

• [object Object]Atlas 训练系列产品[object Object]
• [object Object]Atlas A2 训练系列产品[object Object]
• [object Object]Atlas A3 训练系列产品[object Object]

调用示例