{"id":3079,"date":"2024-04-03T20:05:57","date_gmt":"2024-04-03T15:05:57","guid":{"rendered":"https:\/\/afzalbadshah.com\/?p=3079"},"modified":"2024-05-08T22:21:14","modified_gmt":"2024-05-08T17:21:14","slug":"mpi-with-python-calculating-squares-of-array-elements-using-multiple-processors","status":"publish","type":"post","link":"https:\/\/afzalbadshah.com\/index.php\/2024\/04\/03\/mpi-with-python-calculating-squares-of-array-elements-using-multiple-processors\/","title":{"rendered":"MPI with Python: Calculating Squares of Array Elements Using Multiple Processors"},"content":{"rendered":"\n

In this lab tutorial, we will explore how to utilize multiple processors to compute the squares of elements in an array concurrently using the MPI (Message Passing Interface) library in Python, specifically using the mpi4py module. MPI is a widely-used standard for parallel computing in distributed memory systems. We’ll create a master-worker model where the master process distributes tasks to worker processes, each responsible for computing the square of a subset of the array elements. The detailed tutorial of MPI with a python can be visited here.<\/a> <\/p>\n\n\n\n

Code<\/h3>\n\n\n\n
from mpi4py import MPI\n\ncomm = MPI.COMM_WORLD  # Initialize MPI communicator\nrank = comm.Get_rank()  # Get the rank of the current process (0 for master, 1+ for workers)\nsize = comm.Get_size()  # Get the total number of processes\n\n# Define a simple task for each process (replace with your actual workload)\ndef calculate_square(number):\n    return number * number\n\nif rank == 0:  # Master process\n    data = [2, 4]  # Sample data (modify with your data)\n    for i in range(1, size):  # Send data to worker processes (skip rank 0 - master)\n        comm.send(data[i - 1], dest=i)\n\n    # Receive results from workers\n    results = []\n    for i in range(1, size):\n        result = comm.recv(source=i)\n        results.append(result)\n\n    print(\"Master received results:\", results)\nelse:  # Worker processes (rank 1+)\n    data = comm.recv(source=0)  # Receive data from master\n    result = calculate_square(data)\n    comm.send(result, dest=0)  # Send result back to master\n\nMPI.Finalize()  # Finalize MPI environment\n<\/code><\/pre>\n\n\n\n
\n