From 0e4709daddaf76e71a2de3f7490184453b2c1e17 Mon Sep 17 00:00:00 2001 From: dongdaxiang Date: Sat, 1 Dec 2018 13:14:03 +0800 Subject: [PATCH] add mpi4py helper --- python/paddle/fluid/distributed/helper.py | 20 ++++++++++++++++++++ 1 file changed, 20 insertions(+) create mode 100644 python/paddle/fluid/distributed/helper.py diff --git a/python/paddle/fluid/distributed/helper.py b/python/paddle/fluid/distributed/helper.py new file mode 100644 index 0000000000..8e079b1e8d --- /dev/null +++ b/python/paddle/fluid/distributed/helper.py @@ -0,0 +1,20 @@ +from mpi4py import MPI + +class MPIHelper(object): + def __init__(self): + self.comm = MPI.COMM_WORLD + + def get_rank(self): + return self.comm.Get_rank() + + def get_size(self): + return self.comm.Get_size() + + def get_ip(self): + import socket + local_ip = socket.gethostbyname(socket.gethostname()) + return local_ip + + def get_hostname(self): + import socket + return socket.gethostname() -- GitLab