From 68d5c12b1023a71183be45a44966a9fdd61a19fb Mon Sep 17 00:00:00 2001 From: Shang Zhizhou Date: Mon, 20 Jun 2022 11:10:07 +0800 Subject: [PATCH] fix unittest (#43609) (#43617) --- 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 58d8610ee35..d80bf2ef922 100755 --- a/python/paddle/fluid/tests/unittests/test_activation_op.py +++ b/python/paddle/fluid/tests/unittests/test_activation_op.py @@ -160,7 +160,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(): @@ -168,11 +168,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