qimpy.mpi

MPI communication utilities.

Functions

get_block_slices

Split n_tot tasks into blocks of size block_size.

Classes

BufferView

Expose buffer protocol on torch.Tensor for mpi4py.

ProcessGrid

Process grid of shape dimensions over communicator comm.

TaskDivision

Division of a number of tasks over MPI.

TaskDivisionCustom

Customized division of a number of tasks over MPI.

Waitable

Generic protocol for objects with a wait method.

Waitless

Trivial (identity) Waitable for when result is immediately ready.

Iallreduce_in_place

Perform async Iallreduce in-place on buf.

Modules

qimpy.mpi.globalreduce

Reduction of MPI-distributed tensors to scalars.