mpi4py

mpi4py utilities.

cneuromax.utils.mpi4py.get_mpi_variables()

Retrieves MPI variables from the MPI runtime.

Return type:

tuple[Comm, int, int]

Returns:

• The MPI communicator.

• The rank of the current process.

• The total number of processes.