From 216e085605adcbf89a30cadc076e36dc8186c133 Mon Sep 17 00:00:00 2001
From: lilong12 <lilong12@baidu.com>
Date: Fri, 27 Nov 2020 14:37:40 +0800
Subject: [PATCH] update, test=develop (#29139)

---
 python/paddle/distributed/parallel.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/python/paddle/distributed/parallel.py b/python/paddle/distributed/parallel.py
index 2f951d6aa92..4d60db6f06d 100644
--- a/python/paddle/distributed/parallel.py
+++ b/python/paddle/distributed/parallel.py
@@ -49,7 +49,7 @@ def _start_kv_server(port, http_server_d):
     http_server = KVServer(int(port))
     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