From 2a864c70c42466ea2d8744e73a039badb500b5f5 Mon Sep 17 00:00:00 2001
From: lilong12 <lilong12@baidu.com>
Date: Thu, 26 Nov 2020 17:31:38 +0800
Subject: [PATCH] fix the bug in gloo (#29112)

* update, test=develop
---
 python/paddle/distributed/fleet/base/role_maker.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/python/paddle/distributed/fleet/base/role_maker.py b/python/paddle/distributed/fleet/base/role_maker.py
index b6be992ad1..276d56ea12 100644
--- a/python/paddle/distributed/fleet/base/role_maker.py
+++ b/python/paddle/distributed/fleet/base/role_maker.py
@@ -175,7 +175,8 @@ class Gloo(object):
             http_server = KVServer(port, size_d)
             http_server.start()
             wait_seconds = 5
-            while http_server_d.get("running", False):
+            while http_server_d.get("running",
+                                    False) or not http_server.should_stop():
                 time.sleep(wait_seconds)
             http_server.stop()
 
-- 
GitLab