torch_npu.npu.mstx

函数原型

torch_npu.npu.mstx()

功能说明

打点接口。

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

支持的型号

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