rm mlu ops eager guard tests (#48769)

PaddlePaddle · Dec 9, 2022 · 96e5796 · 96e5796
1 parent 5ecd0ad
commit 96e5796
Show file tree

Hide file tree

Showing 7 changed files with 13 additions and 21 deletions.
diff --git a/python/paddle/fluid/tests/unittests/mlu/test_clip_op_mlu.py b/python/paddle/fluid/tests/unittests/mlu/test_clip_op_mlu.py
@@ -21,7 +21,6 @@
 import paddle.fluid as fluid
 from paddle.fluid import Program, program_guard
 from op_test import OpTest
-from paddle.fluid.framework import _test_eager_guard
 
 paddle.enable_static()
 

diff --git a/python/paddle/fluid/tests/unittests/mlu/test_expand_v2_op_mlu.py b/python/paddle/fluid/tests/unittests/mlu/test_expand_v2_op_mlu.py
@@ -21,7 +21,6 @@
 import paddle.fluid as fluid
 from paddle.fluid import compiler, Program, program_guard
 import paddle
-from paddle.fluid.framework import _test_eager_guard
 
 paddle.enable_static()
 

diff --git a/python/paddle/fluid/tests/unittests/mlu/test_meshgrid_op_mlu.py b/python/paddle/fluid/tests/unittests/mlu/test_meshgrid_op_mlu.py
@@ -21,7 +21,6 @@
 import paddle.fluid as fluid
 import paddle
 from paddle.fluid import compiler, Program, program_guard, core
-from paddle.fluid.framework import _test_eager_guard
 
 paddle.enable_static()
 

diff --git a/python/paddle/fluid/tests/unittests/mlu/test_one_hot_v2_op_mlu.py b/python/paddle/fluid/tests/unittests/mlu/test_one_hot_v2_op_mlu.py
@@ -23,7 +23,7 @@
 import paddle.fluid as fluid
 import paddle.fluid.core as core
 import paddle.fluid.framework as framework
-from paddle.fluid.framework import Program, program_guard, _test_eager_guard
+from paddle.fluid.framework import Program, program_guard
 
 paddle.enable_static()
 
@@ -203,9 +203,8 @@ def test_api_with_dygraph(self):
             one_hot_label = paddle.nn.functional.one_hot(
                 fluid.dygraph.to_variable(label), depth
             )
-            # with _test_eager_guard():
-            #     one_hot_label = paddle.nn.functional.one_hot(
-            #         paddle.to_tensor(label), depth)
+            # one_hot_label = paddle.nn.functional.one_hot(
+            #     paddle.to_tensor(label), depth)
 
     def _run(self, depth):
         label = fluid.layers.data(name="label", shape=[1], dtype="int64")

diff --git a/python/paddle/fluid/tests/unittests/mlu/test_truncated_gaussian_random_op_mlu.py b/python/paddle/fluid/tests/unittests/mlu/test_truncated_gaussian_random_op_mlu.py
@@ -24,7 +24,6 @@
 from op_test import OpTest
 from paddle.fluid.op import Operator
 from paddle.fluid.executor import Executor
-from paddle.fluid.framework import _test_eager_guard
 
 paddle.enable_static()
 
@@ -76,17 +75,16 @@ def gaussian_random_test(self, place):
     # directly
     def gaussian_random_test_eager(self, place):
         with fluid.dygraph.guard(place):
-            with _test_eager_guard():
-                out = paddle._C_ops.truncated_gaussian_random(
-                    self.attrs["shape"],
-                    self.attrs["mean"],
-                    self.attrs["std"],
-                    self.attrs["seed"],
-                    core.VarDesc.VarType.FP32,
-                    place,
-                )
-                self.assertAlmostEqual(numpy.mean(out.numpy()), 0.0, delta=0.1)
-                self.assertAlmostEqual(numpy.var(out.numpy()), 0.773, delta=0.1)
+            out = paddle._C_ops.truncated_gaussian_random(
+                self.attrs["shape"],
+                self.attrs["mean"],
+                self.attrs["std"],
+                self.attrs["seed"],
+                core.VarDesc.VarType.FP32,
+                place,
+            )
+            self.assertAlmostEqual(numpy.mean(out.numpy()), 0.0, delta=0.1)
+            self.assertAlmostEqual(numpy.var(out.numpy()), 0.773, delta=0.1)
 
 
 if __name__ == "__main__":

diff --git a/python/paddle/fluid/tests/unittests/mlu/test_where_op_mlu.py b/python/paddle/fluid/tests/unittests/mlu/test_where_op_mlu.py
@@ -25,7 +25,6 @@
 from paddle.fluid import compiler, Program, program_guard
 from paddle.fluid.op import Operator
 from paddle.fluid.backward import append_backward
-from paddle.fluid.framework import _test_eager_guard
 
 paddle.enable_static()
 

diff --git a/python/paddle/fluid/tests/unittests/mlu/test_yolo_box_op_mlu.py b/python/paddle/fluid/tests/unittests/mlu/test_yolo_box_op_mlu.py
@@ -23,7 +23,6 @@
 import paddle.fluid as fluid
 from paddle.fluid.op import Operator
 from paddle.fluid.executor import Executor
-from paddle.fluid.framework import _test_eager_guard
 
 paddle.enable_static()