optimize test_async_ssa_graph_executor_mnist test=develop

e92ad8a2 · Qiao Longfei · b2c082c4 · e92ad8a2
显示空白变更内容
内联并排

Showing with 2 addition and 2 deletion

python/paddle/fluid/tests/unittests/test_async_ssa_graph_executor_mnist.py ...id/tests/unittests/test_async_ssa_graph_executor_mnist.py +2 -2

未找到文件。
--- 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__":