rm unittests eager guard tests part5 dataloader2dygraph_mnist (#48816)

Author: yjjiang11

python/paddle/fluid/tests/unittests/test_dataloader_dataset.py

@@ -17,12 +17,11 @@
 
 import paddle
 import paddle.vision.transforms as transforms
-from paddle.fluid.framework import _test_eager_guard
 from paddle.io import Dataset
 
 
 class TestDatasetAbstract(unittest.TestCase):
-    def func_test_main(self):
+    def test_main(self):
         dataset = Dataset()
         try:
             d = dataset[0]
@@ -36,11 +35,6 @@ def func_test_main(self):
         except NotImplementedError:
             pass
 
-    def test_main(self):
-        with _test_eager_guard():
-            self.func_test_main()
-        self.func_test_main()
-
 
 class TestDatasetWithDiffOutputPlace(unittest.TestCase):
     def get_dataloader(self, num_workers):
@@ -68,7 +62,7 @@ def run_check_on_cpu(self):
             self.assertTrue(label.place.is_cpu_place())
             break
 
-    def func_test_single_process(self):
+    def test_single_process(self):
         self.run_check_on_cpu()
         if paddle.is_compiled_with_cuda():
             # Get (image, label) tuple from MNIST dataset
@@ -80,12 +74,7 @@ def func_test_single_process(self):
                 self.assertTrue(label.place.is_cuda_pinned_place())
                 break
 
-    def test_single_process(self):
-        with _test_eager_guard():
-            self.func_test_single_process()
-        self.func_test_single_process()
-
-    def func_test_multi_process(self):
+    def test_multi_process(self):
         # DataLoader with multi-process mode is not supported on MacOs and Windows currently
         if sys.platform != 'darwin' and sys.platform != 'win32':
             self.run_check_on_cpu()
@@ -99,11 +88,6 @@ def func_test_multi_process(self):
                     self.assertTrue(label.place.is_cuda_pinned_place())
                     break
 
-    def test_multi_process(self):
-        with _test_eager_guard():
-            self.func_test_multi_process()
-        self.func_test_multi_process()
-
 
 if __name__ == '__main__':
     unittest.main()

python/paddle/fluid/tests/unittests/test_deform_conv2d.py

@@ -19,7 +19,6 @@
 
 import paddle
 import paddle.nn.initializer as I
-from paddle.fluid.framework import _test_eager_guard
 
 
 class TestDeformConv2D(TestCase):
@@ -233,10 +232,6 @@ def test_identity(self):
             self.place = paddle.CUDAPlace(0)
             self._test_identity()
 
-    def test_identity_with_eager_guard(self):
-        with _test_eager_guard():
-            self.test_identity()
-
 
 class TestDeformConv2DFunctional(TestCase):
     batch_size = 4
@@ -544,10 +539,6 @@ def test_identity(self):
             self.place = paddle.CUDAPlace(0)
             self._test_identity()
 
-    def test_identity_with_eager_guard(self):
-        with _test_eager_guard():
-            self.test_identity()
-
 
 # testcases for DeformConv2D
 class TestDeformConv2DWithPadding(TestDeformConv2D):

python/paddle/fluid/tests/unittests/test_deformable_conv_op.py

@@ -18,7 +18,6 @@
 from op_test import OpTest
 
 import paddle
-from paddle.fluid.framework import _test_eager_guard
 
 paddle.enable_static()
 
@@ -442,10 +441,6 @@ def test_invalid_filter():
 
         self.assertRaises(ValueError, test_invalid_filter)
 
-    def test_error_with_eager_guard(self):
-        with _test_eager_guard():
-            self.test_error()
-
 
 class TestDeformConv2DAPI(unittest.TestCase):
     def test_api(self):
@@ -484,10 +479,6 @@ def test_deform_conv2d_v2():
 
         test_deform_conv2d_v2()
 
-    def test_api_with_eager_guard(self):
-        with _test_eager_guard():
-            self.test_api()
-
 
 if __name__ == '__main__':
     unittest.main()

python/paddle/fluid/tests/unittests/test_determinant_op.py

@@ -18,7 +18,6 @@
 from op_test import OpTest
 
 import paddle
-from paddle.fluid.framework import _test_eager_guard
 
 paddle.enable_static()
 
@@ -87,10 +86,6 @@ def test_api_dygraph(self):
         np.testing.assert_allclose(out.numpy(), out_ref, rtol=0.001)
         paddle.enable_static()
 
-    def test_eager(self):
-        with _test_eager_guard():
-            self.test_api_dygraph()
-
 
 class TestSlogDeterminantOp(OpTest):
     def setUp(self):

python/paddle/fluid/tests/unittests/test_diag_v2.py

@@ -20,7 +20,6 @@
 import paddle
 import paddle.fluid as fluid
 from paddle.fluid import Program, program_guard
-from paddle.fluid.framework import _test_eager_guard
 
 
 class TestDiagV2Op(OpTest):
@@ -281,8 +280,6 @@ def run_static(self, use_gpu=False):
     def test_cpu(self):
         paddle.disable_static(place=paddle.fluid.CPUPlace())
         self.run_imperative()
-        with _test_eager_guard():
-            self.run_imperative()
 
         paddle.enable_static()
 
@@ -295,8 +292,6 @@ def test_gpu(self):
 
         paddle.disable_static(place=paddle.fluid.CUDAPlace(0))
         self.run_imperative()
-        with _test_eager_guard():
-            self.run_imperative()
         paddle.enable_static()
 
         with fluid.program_guard(fluid.Program()):

python/paddle/fluid/tests/unittests/test_diagonal_op.py

@@ -18,7 +18,6 @@
 from op_test import OpTest
 
 import paddle
-from paddle.fluid.framework import _test_eager_guard
 
 paddle.enable_static()
 
@@ -157,12 +156,11 @@ def test_api_dygraph(self):
 
     def test_api_eager(self):
         paddle.disable_static(self.place)
-        with _test_eager_guard():
-            x_tensor = paddle.to_tensor(self.x)
-            out = paddle.diagonal(x_tensor)
-            out2 = paddle.diagonal(x_tensor, offset=0, axis1=2, axis2=1)
-            out3 = paddle.diagonal(x_tensor, offset=1, axis1=0, axis2=1)
-            out4 = paddle.diagonal(x_tensor, offset=0, axis1=1, axis2=2)
+        x_tensor = paddle.to_tensor(self.x)
+        out = paddle.diagonal(x_tensor)
+        out2 = paddle.diagonal(x_tensor, offset=0, axis1=2, axis2=1)
+        out3 = paddle.diagonal(x_tensor, offset=1, axis1=0, axis2=1)
+        out4 = paddle.diagonal(x_tensor, offset=0, axis1=1, axis2=2)
         out_ref = np.diagonal(self.x)
         np.testing.assert_allclose(out.numpy(), out_ref, rtol=1e-08)
         out2_ref = np.diagonal(self.x, offset=0, axis1=2, axis2=1)
@@ -174,10 +172,6 @@ def test_api_eager(self):
 
         paddle.enable_static()
 
-    def test_api_eager_dygraph(self):
-        with _test_eager_guard():
-            self.test_api_dygraph()
-
 
 if __name__ == '__main__':
     unittest.main()

python/paddle/fluid/tests/unittests/test_diff_op.py

@@ -19,7 +19,6 @@
 import paddle
 import paddle.fluid as fluid
 import paddle.fluid.core as core
-from paddle.fluid.framework import _test_eager_guard
 
 
 class TestDiffOp(unittest.TestCase):
@@ -75,9 +74,6 @@ def func_dygraph(self):
             self.assertTrue((out.numpy() == self.output).all(), True)
 
     def test_dygraph(self):
-        with _test_eager_guard():
-            self.setUp()
-            self.func_dygraph()
         self.setUp()
         self.func_dygraph()
 
@@ -145,9 +141,6 @@ def func_grad(self):
                 raise RuntimeError("Check Diff Gradient Failed")
 
     def test_grad(self):
-        with _test_eager_guard():
-            self.setUp()
-            self.func_grad()
         self.setUp()
         self.func_grad()
 

python/paddle/fluid/tests/unittests/test_digamma_op.py

@@ -21,7 +21,6 @@
 import paddle
 import paddle.fluid as fluid
 import paddle.static as static
-from paddle.fluid.framework import _test_eager_guard
 
 
 class TestDigammaOp(OpTest):
@@ -95,10 +94,6 @@ def test_in_dynamic_mode(self):
                     res = paddle.digamma(input_t).numpy()
                     np.testing.assert_allclose(res, sc_res, rtol=1e-05)
 
-    def test_in_eager_dynamic_mode(self):
-        with _test_eager_guard():
-            self.test_in_dynamic_mode()
-
     def test_name_argument(self):
         with static.program_guard(static.Program()):
             x = static.data(name="x", shape=self._shape, dtype=self.dtypes[0])
@@ -119,13 +114,6 @@ def test_dtype_error(self):
                 input_t = paddle.to_tensor(input)
                 res = paddle.digamma(input_t)
 
-        with self.assertRaises(RuntimeError):
-            with fluid.dygraph.guard():
-                with _test_eager_guard():
-                    input = np.random.random(self._shape).astype("int32")
-                    input_t = paddle.to_tensor(input)
-                    res = paddle.digamma(input_t)
-
 
 if __name__ == "__main__":
     unittest.main()

python/paddle/fluid/tests/unittests/test_dist_base.py

@@ -30,7 +30,6 @@
 import paddle.fluid.dygraph as dygraph
 import paddle.fluid.incubate.fleet.base.role_maker as role_maker
 from paddle.fluid import compiler
-from paddle.fluid.framework import _test_eager_guard
 from paddle.fluid.incubate.fleet.collective import DistributedStrategy, fleet
 
 RUN_STEP = 5
@@ -1718,16 +1717,6 @@ def check_with_place(
         log_name="",
     ):
         if self._dygraph and (self._gloo_mode or self._nccl2_mode):
-            need_envs.update({"FLAGS_enable_eager_mode": "1"})
-            with _test_eager_guard():
-                self.check_with_place_func(
-                    model_file=model_file,
-                    delta=delta,
-                    check_error_log=check_error_log,
-                    need_envs=need_envs,
-                    log_name=log_name,
-                )
-            need_envs.update({"FLAGS_enable_eager_mode": "0"})
             self.check_with_place_func(
                 model_file=model_file,
                 delta=delta,

python/paddle/fluid/tests/unittests/test_dygraph_mnist_fp16.py

@@ -18,7 +18,6 @@
 
 import paddle
 import paddle.fluid as fluid
-from paddle.fluid.framework import _test_eager_guard
 from paddle.nn import Linear
 
 
@@ -136,8 +135,6 @@ def func_mnist_fp16(self):
             print(loss.numpy())
 
     def test_mnist_fp16(self):
-        with _test_eager_guard():
-            self.func_mnist_fp16()
         self.func_mnist_fp16()