Scribd
Upload
11 views4 pages

DS Assignment 3

Assignment 3 ds sppu

Uploaded by

Gaurav Bhagat
Copyright
© © All Rights Reserved
We take content rights seriously. If you suspect this is your content, claim it here.
Available Formats
Download as PDF, TXT or read online on Scribd
11 views4 pages

DS Assignment 3

Assignment 3 ds sppu

Uploaded by

Gaurav Bhagat
Copyright
© © All Rights Reserved
We take content rights seriously. If you suspect this is your content, claim it here.
Available Formats
Download as PDF, TXT or read online on Scribd
You are on page 1/ 4

Code:

ArrSum.java

// client

import mpi.MPI;

import java.util.Scanner;

import mpi.*;

public class ArrSum {

public static void main(String[] args) throws Exception{

MPI.Init(args);

int rank = MPI.COMM_WORLD.Rank();

int size = MPI.COMM_WORLD.Size();

int unitsize = 5;

int root = 0;

int send_buffer[] = null;

// 1 process is expected to handle 4 elements

send_buffer = new int [unitsize * size];

int recieve_buffer[] = new int [unitsize];

int new_recieve_buffer[] = new int [size];

// Set data for distribution

if(rank == root) {

int total_elements = unitsize * size;

System.out.println("Enter " + total_elements + " elements");


for(int i = 0; i < total_elements; i++) {

System.out.println("Element " + i + "\t = " + i);

send_buffer[i] = i;

// Scatter data to processes

MPI.COMM_WORLD.Scatter(

send_buffer,

0,

unitsize,

MPI.INT,

recieve_buffer,

0,

unitsize,

MPI.INT,

root

);

// Calculate sum at non root processes

// Store result in first index of array

for(int i = 1; i < unitsize; i++) {

recieve_buffer[0] += recieve_buffer[i];

System.out.println(

"Intermediate sum at process " + rank + " is " + recieve_buffer[0]

);
// Gather data from processes

MPI.COMM_WORLD.Gather(

recieve_buffer,

0,

1,

MPI.INT,

new_recieve_buffer,

0,

1,

MPI.INT,

root

);

// Aggregate output from all non root processes

if(rank == root) {

int total_sum = 0;

for(int i = 0; i < size; i++) {

total_sum += new_recieve_buffer[i];

System.out.println("Final sum : " + total_sum);

MPI.Finalize();

}}

You might also like