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…
Reference in new issue