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

4.5.1. MPI helper functions


Initialize a global MPI communicator for tkwant.


Free the global tkwant MPI communicator


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.