torch_npu.profiler.supported_activities

接口原型

torch_npu.profiler.supported_activities()

功能描述

查询当前支持采集的activities参数的CPU、NPU事件。

支持的型号

• Atlas 训练系列产品
• Atlas A2 训练系列产品

调用示例

import torch
import torch_npu

...

torch_npu.profiler.supported_activities()