fix the bug in gloo (#29112)

* update, test=develop
4 years ago · 2a864c70c4
parent 7de2db4a81
commit 2a864c70c4
1 changed files with 2 additions and 1 deletions
--- 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()