torch_npu.profiler.supported_activities
接口原型
1 | torch_npu.profiler.supported_activities() |
功能描述
查询当前支持采集的activities参数的CPU、NPU事件。
支持的型号
Atlas 训练系列产品 Atlas A2 训练系列产品 Atlas A3 训练系列产品
调用示例
1 2 3 4 5 6 | import torch import torch_npu ... torch_npu.profiler.supported_activities() |
父主题： torch_npu.profiler