qimpy.mpi

MPI communication utilities.

Functions

all_gather_padded	Gather send with possibly uneven dimension 0 sizes along group.
all_gather_scalars	Gather scalars send along group into an array.
all_reduce_scalars
get_block_slices	Split n_tot tasks into blocks of size block_size.

Classes

ProcessGrid	Process grid of shape dimensions over the global process group.
TaskDivision	Division of a number of tasks over MPI.
TaskDivisionCustom	Customized division of a number of tasks over a process group.
Waitable	Generic protocol for objects with a wait method.
Waitless	Trivial (identity) Waitable for when result is immediately ready.

Modules

globalreduce

Reduction of distributed tensors to scalars.