Source code for aces_test_mpi_install

#!/usr/bin/env python
"""
Parallel Hello World
"""

import sys

from mpi4py import MPI


[docs]def main():
    size = MPI.COMM_WORLD.Get_size()
    rank = MPI.COMM_WORLD.Get_rank()
    name = MPI.Get_processor_name()

    sys.stdout.write(
        "Hello, World! I am process %d of %d on %s.\n" % (rank, size, name)
    )


[docs]def cli():
    import argparse

    parser = argparse.ArgumentParser(help="Run a parallel hello world")
    args = parser.parse_args()
    main()


if __name__ == "__main__":
    cli()