From c6296b2b0ed55c0d224da870338f106401fe786a Mon Sep 17 00:00:00 2001 From: furnace <34057289+windstamp@users.noreply.github.com> Date: Mon, 11 Jan 2021 17:00:29 +0800 Subject: [PATCH] fix empty op unit test fail sometimes (#30225) --- python/paddle/fluid/tests/unittests/test_empty_op.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/python/paddle/fluid/tests/unittests/test_empty_op.py b/python/paddle/fluid/tests/unittests/test_empty_op.py index e8b1f836fc..b8ff66a910 100644 --- a/python/paddle/fluid/tests/unittests/test_empty_op.py +++ b/python/paddle/fluid/tests/unittests/test_empty_op.py @@ -39,7 +39,7 @@ class TestEmptyOp(OpTest): min_value = np.nanmin(outs[0]) always_full_zero = max_value == 0.0 and min_value == 0.0 - always_non_full_zero = max_value > min_value + always_non_full_zero = max_value >= min_value self.assertTrue(always_full_zero or always_non_full_zero, 'always_full_zero or always_non_full_zero.') elif data_type in ['bool']: @@ -124,7 +124,7 @@ class TestEmptyOp_ShapeTensor(OpTest): min_value = np.nanmin(outs[0]) always_full_zero = max_value == 0.0 and min_value == 0.0 - always_non_full_zero = max_value > min_value + always_non_full_zero = max_value >= min_value self.assertTrue(always_full_zero or always_non_full_zero, 'always_full_zero or always_non_full_zero.') elif data_type in ['bool']: @@ -169,7 +169,7 @@ class TestEmptyOp_ShapeTensorList(OpTest): min_value = np.nanmin(outs[0]) always_full_zero = max_value == 0.0 and min_value == 0.0 - always_non_full_zero = max_value > min_value + always_non_full_zero = max_value >= min_value self.assertTrue(always_full_zero or always_non_full_zero, 'always_full_zero or always_non_full_zero.') elif data_type in ['bool']: @@ -186,7 +186,7 @@ class TestEmptyAPI(unittest.TestCase): def __check_out__(self, out, dtype='float32'): max_value = np.nanmax(np.array(out)) min_value = np.nanmin(np.array(out)) - always_non_full_zero = max_value > min_value + always_non_full_zero = max_value >= min_value always_full_zero = max_value == 0.0 and min_value == 0.0 self.assertTrue(always_full_zero or always_non_full_zero, 'always_full_zero or always_non_full_zero.')