async_executor stop add barrier_all & finalize

6 years ago · 5f0358add9
parent 5a4e2e599c
commit 5f0358add9
1 changed files with 2 additions and 0 deletions
--- a/python/paddle/fluid/async_executor.py
+++ b/python/paddle/fluid/async_executor.py
@ -237,6 +237,8 @@ class AsyncExecutor(object):
        if self.instance.is_first_worker():
            self.executor.stop_server()
        self.instance.barrier_worker()  #sync
+        self.instance.barrier_all()
+        self.instance.finalize()

    def init_server(self, dist_desc):
        """