dedalus.tools.parallel

Tools for running in parallel.

Module Contents

class Sync(comm=MPI.COMM_WORLD, enter=True, exit=True)

Context manager for synchronizing MPI processes.

Parameters:

  • enter (boolean, optional) – Apply MPI barrier on entering context. Default: True

  • exit (boolean, optional) – Apply MPI barrier on exiting context. Default: True

class RotateProcesses(comm=MPI.COMM_WORLD)

Context manager for rotating execution between MPI processes.

Parameters:

comm (MPI communicator, optional)

class ProfileWrapper(stats)

Pickleable wrapper for cProfile.Profile for use with pstats.Stats

create_stats()
parallel_mkdir(path, comm=MPI.COMM_WORLD)

Create a directory from root process.