lib.mpi.set_independent_seed

lib.mpi.set_independent_seed(seed=None, mpicomm=None, size=10000)

Set independent global np.random and random seed for all MPI processes.

Parameters
  • seed (int, default=None) – Random seed to use when generating seeds.

  • mpicomm (MPI communicator, default=None) – Communicator to use for broadcasting. Defaults to current communicator.

  • size (int, default=10000) – Number of seeds to be generated. To ensure random draws are independent of the number of ranks, this should be larger than the total number of processes that will ever be used.

Returns

seed – Seed used to initialize np.random and random global states.

Return type

int