torch_npu.npu.mstx

API接口

1

torch_npu.npu.mstx()

功能描述

打点接口。

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

支持的型号

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