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