diff --git a/python/paddle/fluid/tests/unittests/test_pool2d_int8_mkldnn_op.py b/python/paddle/fluid/tests/unittests/test_pool2d_int8_mkldnn_op.py index 954d9993b29622f6a4806c8c8dd45f21abafc4fa..e73ac7c0aa51a44a91641a6fcc42780be49c08c9 100644 --- a/python/paddle/fluid/tests/unittests/test_pool2d_int8_mkldnn_op.py +++ b/python/paddle/fluid/tests/unittests/test_pool2d_int8_mkldnn_op.py @@ -148,7 +148,7 @@ class TestPool2D_Op(OpTest): self.outputs = {'Out': output} def test_check_output(self): - self.check_output() + self.check_output_with_place(core.CPUPlace(), atol=1e-5) def init_test_case(self): self.shape = [2, 3, 5, 5]