optimize test_async_ssa_graph_executor_mnist test=develop

6 years ago · e92ad8a209
parent b2c082c46f
commit e92ad8a209
1 changed files with 2 additions and 2 deletions
--- a/python/paddle/fluid/tests/unittests/test_async_ssa_graph_executor_mnist.py
+++ b/python/paddle/fluid/tests/unittests/test_async_ssa_graph_executor_mnist.py
@ -178,8 +178,8 @@ class TestAsyncSSAGraphExecutor(unittest.TestCase):
                        main_program=fluid.Program(),
                        startup_program=fluid.Program()):
                    test()
-        assert int(step_list[0] / 2) == int(step_list[1])
-        assert int(step_list[1] / 2) == int(step_list[2])
+        assert abs(int(step_list[0] / 2) - int(step_list[1])) < 5
+        assert abs(int(step_list[1] / 2) - int(step_list[2])) < 5


 if __name__ == "__main__":