Skip to content

Commit

Permalink
update
Browse files Browse the repository at this point in the history
  • Loading branch information
typhoonzero committed Apr 16, 2018
1 parent 745aacf commit d78cab1
Showing 1 changed file with 5 additions and 4 deletions.
9 changes: 5 additions & 4 deletions python/paddle/fluid/tests/unittests/test_dist_train_op.py
Original file line number Diff line number Diff line change
Expand Up @@ -33,17 +33,18 @@ def test_send(self):
p.daemon = True
p.start()

time.sleep(8)
time.sleep(10)
with open("/tmp/paddle.selected_port", "r") as fn:
selected_port = int(fn.readlines()[0])
self.init_client(place, selected_port)
# FIXME(typhoonzero): find a way to gracefully shutdown the server.
os.system("kill -9 %d" % p.pid)
p.join()

self.run_local(place)
self.assertTrue(numpy.allclose(self.local_out, self.dist_out))

# FIXME(typhoonzero): find a way to gracefully shutdown the server.
os.system("kill -9 %d" % p.pid)
p.join()

def init_serv(self, place):
main = fluid.Program()

Expand Down

0 comments on commit d78cab1

Please sign in to comment.