From c5097af705fbe0de516e352afe13a7c67a4ba19c Mon Sep 17 00:00:00 2001 From: Shang Zhizhou Date: Fri, 17 Jun 2022 14:25:18 +0800 Subject: [PATCH] fix unittest (#43609) --- python/paddle/fluid/tests/unittests/test_activation_op.py | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) diff --git a/python/paddle/fluid/tests/unittests/test_activation_op.py b/python/paddle/fluid/tests/unittests/test_activation_op.py index 7dde0483823..dd242920db3 100755 --- a/python/paddle/fluid/tests/unittests/test_activation_op.py +++ b/python/paddle/fluid/tests/unittests/test_activation_op.py @@ -168,7 +168,7 @@ class TestParameter(object): exe = fluid.Executor(place) result, = exe.run(feed={"X": np_x}, fetch_list=[out]) expected = eval("np.%s(np_x)" % self.op_type) - self.assertEqual(result, expected) + self.assertTrue(np.allclose(result, expected)) def test_dygraph(self): with fluid.dygraph.guard(): @@ -176,11 +176,7 @@ class TestParameter(object): x = fluid.dygraph.to_variable(np_x) z = eval("paddle.%s(x).numpy()" % self.op_type) z_expected = eval("np.%s(np_x)" % self.op_type) - # ROCM platform will fail in assertEqual - if core.is_compiled_with_rocm(): - self.assertTrue(np.allclose(z, z_expected)) - else: - self.assertEqual(z, z_expected) + self.assertTrue(np.allclose(z, z_expected)) class TestSigmoid(TestActivation): -- GitLab