Skip to content

Commit

Permalink
Switch some dy2st UT to eager mode (#41052)
Browse files Browse the repository at this point in the history
* Switch some dy2st UT to eager mode

* Add UT
  • Loading branch information
0x45f committed Mar 30, 2022
1 parent 1042f42 commit a5bfa79
Show file tree
Hide file tree
Showing 7 changed files with 14 additions and 7 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -262,4 +262,5 @@ def init_dygraph_func(self):


if __name__ == '__main__':
unittest.main()
with fluid.framework._test_eager_guard():
unittest.main()
Original file line number Diff line number Diff line change
Expand Up @@ -425,4 +425,5 @@ def test_ast_to_func(self):


if __name__ == '__main__':
unittest.main()
with paddle.fluid.framework._test_eager_guard():
unittest.main()
Original file line number Diff line number Diff line change
Expand Up @@ -421,4 +421,5 @@ def test_ast_to_func(self):


if __name__ == '__main__':
unittest.main()
with fluid.framework._test_eager_guard():
unittest.main()
Original file line number Diff line number Diff line change
Expand Up @@ -91,4 +91,5 @@ def train(self, to_static=False):


if __name__ == '__main__':
unittest.main()
with paddle.fluid.framework._test_eager_guard():
unittest.main()
Original file line number Diff line number Diff line change
Expand Up @@ -108,4 +108,5 @@ def train(self, to_static=False):


if __name__ == '__main__':
unittest.main()
with paddle.fluid.framework._test_eager_guard():
unittest.main()
Original file line number Diff line number Diff line change
Expand Up @@ -213,4 +213,5 @@ def test_train(self):


if __name__ == '__main__':
unittest.main()
with fluid.framework._test_eager_guard():
unittest.main()
Original file line number Diff line number Diff line change
Expand Up @@ -170,4 +170,5 @@ def test_dygraph_static_same_loss(self):


if __name__ == '__main__':
unittest.main()
with fluid.framework._test_eager_guard():
unittest.main()

0 comments on commit a5bfa79

Please sign in to comment.