#! /usr/bin/env python

from mpi4py import MPI


comm = MPI.COMM_WORLD


def get_host():
    return MPI.Get_processor_name()


def get_rank():
    return comm.Get_rank()


def finalize():
    return MPI.Finalize()

#! /usr/bin/env python

from mpi4py import MPI


comm = MPI.COMM_WORLD


def get_host():
    return MPI.Get_processor_name()


def get_rank():
    return comm.Get_rank()


def host_rank_mapping():
    """Get host to rank mapping

    Return dictionary mapping ranks to host
    """
    d = {}
    for (host, rank) in comm.allgather((get_host(), get_rank())):
        if host not in d:
            d[host] = []
        d[host].append(rank)
    return d


def finalize():
    return MPI.Finalize()

Add function to get the host-rank mapping of mpi tasks
Add function to get the host-rank mapping of mpi tasks
