From bb28eab8332fa533c7d4ae6ffa50eb8cc60b8712 Mon Sep 17 00:00:00 2001 From: chenguowei01 Date: Wed, 3 Aug 2022 08:09:10 +0000 Subject: [PATCH] update test_uniform_random_inplace_op.py --- .../test_uniform_random_inplace_op.py | 31 +++---------------- 1 file changed, 5 insertions(+), 26 deletions(-) diff --git a/python/paddle/fluid/tests/unittests/test_uniform_random_inplace_op.py b/python/paddle/fluid/tests/unittests/test_uniform_random_inplace_op.py index ae772818cbc6d..d677d3b34c64c 100644 --- a/python/paddle/fluid/tests/unittests/test_uniform_random_inplace_op.py +++ b/python/paddle/fluid/tests/unittests/test_uniform_random_inplace_op.py @@ -170,7 +170,7 @@ class TestUniformRandomInplaceGrad(unittest.TestCase): def setUp(self): self.shape = (1000, 784) - def test_uniform_random_inplace_grad(self): + def run_(self): fluid.set_flags({"FLAGS_retain_grad_for_all_tensor": True}) def test_grad(): @@ -191,33 +191,12 @@ def test_grad(): test_grad() fluid.set_flags({"FLAGS_retain_grad_for_all_tensor": False}) - -class TestUniformRandomInplaceGradOldDygraph(unittest.TestCase): - - def setUp(self): - self.shape = (1000, 784) - def test_uniform_random_inplace_grad(self): - _enable_legacy_dygraph() - fluid.set_flags({"FLAGS_retain_grad_for_all_tensor": True}) - - def test_grad(): - tensor_a = paddle.ones(self.shape) - tensor_a.stop_gradient = False - tensor_b = tensor_a * 0.5 - tensor_b.uniform_(min=-2, max=2) - loss = tensor_b.sum() - loss.backward() - uniform_grad = tensor_b.grad.numpy() - self.assertTrue((uniform_grad == 0).all()) + self.run_() - places = ['cpu'] - if fluid.core.is_compiled_with_cuda(): - places.append('gpu') - for place in places: - paddle.set_device(place) - test_grad() - fluid.set_flags({"FLAGS_retain_grad_for_all_tensor": False}) + def test_uniform_random_inplace_grad_old_dygraph(self): + _enable_legacy_dygraph() + self.run_() _disable_legacy_dygraph()