cosmoslik.mpi package¶
Submodules¶
-
cosmoslik.mpi.mpi.
flatten
(l)¶ Returns a list of lists joined into one
-
cosmoslik.mpi.mpi.
get_mpi
()¶ Return (rank,size,comm)
-
cosmoslik.mpi.mpi.
get_pool
()¶ Gets a pool object which can be passed to things that expect it to have a pool.map function.
-
cosmoslik.mpi.mpi.
mpi_consistent
(value)¶ Returns the value that the root process provided.
-
cosmoslik.mpi.mpi.
mpi_map
(function, sequence, distribute=True)¶ A map function parallelized with MPI. If this program was called with mpiexec -n $NUM, then partitions the sequence into $NUM blocks and each MPI process does the rank-th one. Note: If this function is called recursively, only the first call will be parallelized
Keyword arguments: distribute – If true, every process receives the answer
otherwise only the root process does (default=True)
-
cosmoslik.mpi.mpi.
partition
(l, n)¶ Partition l into n nearly equal sublists