torch_npu.npu.get_npu_overflow_flag

API接口

torch_npu.npu.get_npu_overflow_flag()

功能描述

检测npu计算过程中是否有数值溢出。

支持的型号

• Atlas 训练系列产品

示例

>>>a = torch.Tensor([65535]).npu().half()
>>>a = a + a
>>>ret = torch_npu.npu.get_npu_overflow_flag()