torch_npu.npu.get_npu_overflow_flag()
检测NPU计算过程中是否有数值溢出。
Atlas 训练系列产品
1 2 3
>>>a = torch.Tensor([65535]).npu().half() >>>a = a + a >>>ret = torch_npu.npu.get_npu_overflow_flag()