def set_atol(self):
    self.atol = 1e-10
    self.rtol = 1e-06
    self.atol_fp16 = 0.001
    self.rtol_fp16 = 0.001
def set_atol(self):
    self.atol = 1e-10
    self.rtol = 1e-06
    self.atol_fp16 = 0.001
    self.rtol_fp16 = 0.001