torch_npu.npu.get_npu_overflow_flag()
检测npu计算过程中是否有数值溢出。
>>>a = torch.Tensor([65535]).npu().half() >>>a = a + a >>>ret = torch_npu.npu.get_npu_overflow_flag()