From 2003610e325926a1132ec7117f42fd92f8653ab2 Mon Sep 17 00:00:00 2001
From: 0x45f <23097963+0x45f@users.noreply.github.com>
Date: Thu, 31 Mar 2022 19:11:27 +0800
Subject: [PATCH] Switch some dy2st UT to eager (#41175)

---
 .../fluid/tests/unittests/dygraph_to_static/test_cycle_gan.py  | 3 ++-
 .../paddle/fluid/tests/unittests/dygraph_to_static/test_lac.py | 3 ++-
 .../fluid/tests/unittests/dygraph_to_static/test_resnet_amp.py | 3 ++-
 .../tests/unittests/dygraph_to_static/test_resnet_pure_fp16.py | 3 ++-
 4 files changed, 8 insertions(+), 4 deletions(-)

diff --git a/python/paddle/fluid/tests/unittests/dygraph_to_static/test_cycle_gan.py b/python/paddle/fluid/tests/unittests/dygraph_to_static/test_cycle_gan.py
index d6840ed628..8a9a1e1920 100644
--- a/python/paddle/fluid/tests/unittests/dygraph_to_static/test_cycle_gan.py
+++ b/python/paddle/fluid/tests/unittests/dygraph_to_static/test_cycle_gan.py
@@ -631,4 +631,5 @@ class TestCycleGANModel(unittest.TestCase):
 
 
 if __name__ == "__main__":
-    unittest.main()
+    with fluid.framework._test_eager_guard():
+        unittest.main()
diff --git a/python/paddle/fluid/tests/unittests/dygraph_to_static/test_lac.py b/python/paddle/fluid/tests/unittests/dygraph_to_static/test_lac.py
index 88c6060abf..19965821e8 100644
--- a/python/paddle/fluid/tests/unittests/dygraph_to_static/test_lac.py
+++ b/python/paddle/fluid/tests/unittests/dygraph_to_static/test_lac.py
@@ -610,4 +610,5 @@ class TestLACModel(unittest.TestCase):
 
 
 if __name__ == "__main__":
-    unittest.main()
+    with fluid.framework._test_eager_guard():
+        unittest.main()
diff --git a/python/paddle/fluid/tests/unittests/dygraph_to_static/test_resnet_amp.py b/python/paddle/fluid/tests/unittests/dygraph_to_static/test_resnet_amp.py
index 509accd8f8..1d45e906cd 100644
--- a/python/paddle/fluid/tests/unittests/dygraph_to_static/test_resnet_amp.py
+++ b/python/paddle/fluid/tests/unittests/dygraph_to_static/test_resnet_amp.py
@@ -116,4 +116,5 @@ class TestResnet(unittest.TestCase):
 
 
 if __name__ == '__main__':
-    unittest.main()
+    with fluid.framework._test_eager_guard():
+        unittest.main()
diff --git a/python/paddle/fluid/tests/unittests/dygraph_to_static/test_resnet_pure_fp16.py b/python/paddle/fluid/tests/unittests/dygraph_to_static/test_resnet_pure_fp16.py
index cf5c2b7311..49d114730e 100644
--- a/python/paddle/fluid/tests/unittests/dygraph_to_static/test_resnet_pure_fp16.py
+++ b/python/paddle/fluid/tests/unittests/dygraph_to_static/test_resnet_pure_fp16.py
@@ -125,4 +125,5 @@ class TestResnet(unittest.TestCase):
 
 
 if __name__ == '__main__':
-    unittest.main()
+    with fluid.framework._test_eager_guard():
+        unittest.main()
-- 
GitLab