@@ -150,9 +150,7 @@ def setUp(self):
150
150
}
151
151
152
152
def test_check_output (self ):
153
- self .check_output_with_place (
154
- core .CUDAPlace (0 ), atol = 1e-5 , check_pir = True
155
- )
153
+ self .check_output_with_place (core .CUDAPlace (0 ), check_pir = True )
156
154
157
155
def test_check_grad (self ):
158
156
self .check_grad_with_place (
@@ -186,9 +184,7 @@ def init_dtype(self):
186
184
self .dtype = np .float16
187
185
188
186
def test_check_output (self ):
189
- self .check_output_with_place (
190
- core .CUDAPlace (0 ), atol = 5e-2 , check_pir = True
191
- )
187
+ self .check_output_with_place (core .CUDAPlace (0 ), check_pir = True )
192
188
193
189
def test_check_grad (self ):
194
190
self .check_grad_with_place (
@@ -274,9 +270,7 @@ def setUp(self):
274
270
}
275
271
276
272
def test_check_output (self ):
277
- self .check_output_with_place (
278
- core .CUDAPlace (0 ), atol = 5e-2 , check_pir = True
279
- )
273
+ self .check_output_with_place (core .CUDAPlace (0 ), check_pir = True )
280
274
281
275
def test_check_grad (self ):
282
276
self .check_grad_with_place (
@@ -314,9 +308,7 @@ def init_loss_params(self):
314
308
class TestMarginCrossEntropyOpCPU (TestMarginCrossEntropyOp ):
315
309
def test_check_output (self ):
316
310
try :
317
- self .check_output_with_place (
318
- core .CPUPlace (), atol = 1e-5 , check_pir = True
319
- )
311
+ self .check_output_with_place (core .CPUPlace (), check_pir = True )
320
312
except RuntimeError :
321
313
pass
322
314
0 commit comments