4.5. tkwant.mpi – Parallelizing simulations with the Message Passing Interface (MPI)

4.5.1. MPI helper functions

communicator_init([comm])

Initialize a global MPI communicator for tkwant.

communicator_free()

Free the global tkwant MPI communicator

get_communicator([comm])

Return the global tkwant MPI communicator if input comm is None

distribute_dict(input_dict, distribute_keys, …)

Distribute a dict over all MPI ranks.

DistributedDict([data, comm])

A class to handle a dictionary which is distributed over MPI ranks.

round_robin(comm, key)

Schedule integer keys in a round robin fashion to the MPI ranks.