torch_npu.npu.mstx

产品支持情况

产品	是否支持
Atlas A3 训练系列产品	√
Atlas A3 推理系列产品	√
Atlas A2 训练系列产品	√
Atlas A2 推理系列产品	√
Atlas 推理系列产品	√
Atlas 训练系列产品	√

功能说明

打点接口。

用于为torch_npu.profiler._ExperimentalConfig的mstx提供打点接口调用。

函数原型

torch_npu.npu.mstx()

参数说明

无

返回值说明

无

调用示例

以下是关键步骤的代码示例，不可直接拷贝编译运行，仅供参考。

import torch
import torch_npu
mstx_object = torch_npu.npu.mstx()