Source code for ASI_Embedding.parallel_utils

from mpi4py import MPI

[docs] def root_print(*args, **kwargs): "Prints only from the root node" rank = MPI.COMM_WORLD.Get_rank() if rank == 0: print(*args, **kwargs)
# TODO: Dynamic printing routine for given tasks/all tasks. #def parprint(*args, **kwargs): # "Prints on given task, list of tasks, for all tasks" # # rank = MPI.COMM_WORLD.Get_rank()