def _make_new_comm(rank_list): import distarray.localapi.mpiutils as mpiutils new_comm = mpiutils.create_comm_with_list(rank_list) if not mpiutils.get_base_comm(): mpiutils.set_base_comm(new_comm) return proxyize(new_comm) # noqa
def local_test_type(): from mpi4py.MPI import Intracomm from distarray.localapi.mpiutils import get_base_comm bc = get_base_comm() return isinstance(bc, Intracomm), bc.rank, bc.size
def local_test_type(): from mpi4py.MPI import Intracomm from distarray.localapi.mpiutils import get_base_comm bc = get_base_comm() return isinstance(bc, Intracomm), bc.rank, bc.size
def _make_new_comm(rank_list): import distarray.localapi.mpiutils as mpiutils new_comm = mpiutils.create_comm_with_list(rank_list) if not mpiutils.get_base_comm(): mpiutils.set_base_comm(new_comm) return proxyize(new_comm) # noqa