torch_npu.profiler.supported_ai_core_metrics

接口原型

torch_npu.profiler.supported_ai_core_metrics()

功能描述

查询当前支持的torch_npu.profiler.AiCMetrics的AI Core性能指标采集项。

支持的型号

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

调用示例

import torch
import torch_npu

...

torch_npu.profiler.supported_ai_core_metrics()