diff --git a/d2o/random.py b/d2o/random.py new file mode 100644 index 0000000000000000000000000000000000000000..430015246bd9519d822ecea82bac74b998bc3d73 --- /dev/null +++ b/d2o/random.py @@ -0,0 +1,15 @@ +# -*- coding: utf-8 -*- + +import numpy as np + +from d2o.config import configuration as gc,\ + dependency_injector as gdi + +MPI = gdi[gc['mpi_module']] +comm = gc['default_comm'] +rank = comm.rank + + +def seed(seed=None): + seed = rank if seed is None else np.uint32(seed) + rank + np.random.seed(seed)