From 5f0358add9767390b8bc329c97236f8d72ce758e Mon Sep 17 00:00:00 2001
From: heqiaozhi <heqiaozhi@baidu.com>
Date: Mon, 17 Dec 2018 16:18:10 +0800
Subject: [PATCH] async_executor stop add barrier_all & finalize

---
 python/paddle/fluid/async_executor.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/python/paddle/fluid/async_executor.py b/python/paddle/fluid/async_executor.py
index bd3213865..3181654fe 100644
--- 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):
         """
-- 
GitLab