add mpi4py helper

revert-15207-remove_op_handle_lock_and_fix_var
dongdaxiang 6 years ago
parent 41e19eb431
commit 0e4709dadd

@ -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()
Loading…
Cancel
Save