Source code for hello_mpi_world
#!/usr/bin/env python
"""
Parallel Hello World
"""
from __future__ import annotations
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()
msg = f"Hello, World! I am process {rank} of {size} on {name}.\n"
sys.stdout.write(msg)
[docs]
def cli():
import argparse
parser = argparse.ArgumentParser(description="Run a parallel hello world")
_ = parser.parse_args()
main()
if __name__ == "__main__":
cli()