mpi Module

class mpi.MDMPI

Bases: object

Class to store a MPI communicator such that it can be used everywhere (bottom level of hierarchy).

barrier()

alias to comm barrier method.

comm

Return the current communicator.

create_cart(dims, periods, reorder_flag)

Create an mpi cart on the current comm

dims

Return the current dimensions.

fortran_comm
get_move_send_recv_ranks()
nproc

Return the current size.

periods

Return the current periods.

print_str(*args)

Method to print on rank 0 to stdout

query_boundary_exist

Return for each direction: Flag if process is a boundary edge or interior edge 1 or 0.

Xl 0, Xu 1 Yl 2, Yu 3 Zl 4, Zu 5

query_halo_exist

Return for each direction: Flag if process has a halo on each face.

Xl 0, Xu 1 Yl 2, Yu 3 Zl 4, Zu 5

rank

Return the current rank.

set_periods(p=None)

set periods (if for some reason mpi4py does not set these this prives a soln.

shift(offset=(0, 0, 0), ignore_periods=False)

Returns rank of process found at a given offset, will return -1 if no process exists.

Parameters:offset (tuple) – 3-tuple offset from current process.
top

Return the current topology.

mpi.all_reduce(array)
mpi.decomposition

alias of Enum

mpi.enum(**enums)
mpi.reset_mpi()

Previous topic

method Module

Next topic

pairloop Module

This Page