Blame examples/cpi.c

Packit 0848f5
/* -*- Mode: C; c-basic-offset:4 ; indent-tabs-mode:nil ; -*- */
Packit 0848f5
/*
Packit 0848f5
 *  (C) 2001 by Argonne National Laboratory.
Packit 0848f5
 *      See COPYRIGHT in top-level directory.
Packit 0848f5
 */
Packit 0848f5
Packit 0848f5
#include "mpi.h"
Packit 0848f5
#include <stdio.h>
Packit 0848f5
#include <math.h>
Packit 0848f5
Packit 0848f5
double f(double);
Packit 0848f5
Packit 0848f5
double f(double a)
Packit 0848f5
{
Packit 0848f5
    return (4.0 / (1.0 + a*a));
Packit 0848f5
}
Packit 0848f5
Packit 0848f5
int main(int argc,char *argv[])
Packit 0848f5
{
Packit 0848f5
    int    n, myid, numprocs, i;
Packit 0848f5
    double PI25DT = 3.141592653589793238462643;
Packit 0848f5
    double mypi, pi, h, sum, x;
Packit 0848f5
    double startwtime = 0.0, endwtime;
Packit 0848f5
    int    namelen;
Packit 0848f5
    char   processor_name[MPI_MAX_PROCESSOR_NAME];
Packit 0848f5
Packit 0848f5
    MPI_Init(&argc,&argv);
Packit 0848f5
    MPI_Comm_size(MPI_COMM_WORLD,&numprocs);
Packit 0848f5
    MPI_Comm_rank(MPI_COMM_WORLD,&myid);
Packit 0848f5
    MPI_Get_processor_name(processor_name,&namelen);
Packit 0848f5
Packit 0848f5
    fprintf(stdout,"Process %d of %d is on %s\n",
Packit 0848f5
	    myid, numprocs, processor_name);
Packit 0848f5
    fflush(stdout);
Packit 0848f5
Packit 0848f5
    n = 10000;			/* default # of rectangles */
Packit 0848f5
    if (myid == 0)
Packit 0848f5
	startwtime = MPI_Wtime();
Packit 0848f5
Packit 0848f5
    MPI_Bcast(&n, 1, MPI_INT, 0, MPI_COMM_WORLD);
Packit 0848f5
Packit 0848f5
    h   = 1.0 / (double) n;
Packit 0848f5
    sum = 0.0;
Packit 0848f5
    /* A slightly better approach starts from large i and works back */
Packit 0848f5
    for (i = myid + 1; i <= n; i += numprocs)
Packit 0848f5
    {
Packit 0848f5
	x = h * ((double)i - 0.5);
Packit 0848f5
	sum += f(x);
Packit 0848f5
    }
Packit 0848f5
    mypi = h * sum;
Packit 0848f5
Packit 0848f5
    MPI_Reduce(&mypi, &pi, 1, MPI_DOUBLE, MPI_SUM, 0, MPI_COMM_WORLD);
Packit 0848f5
Packit 0848f5
    if (myid == 0) {
Packit 0848f5
	endwtime = MPI_Wtime();
Packit 0848f5
	printf("pi is approximately %.16f, Error is %.16f\n",
Packit 0848f5
	       pi, fabs(pi - PI25DT));
Packit 0848f5
	printf("wall clock time = %f\n", endwtime-startwtime);	       
Packit 0848f5
	fflush(stdout);
Packit 0848f5
    }
Packit 0848f5
Packit 0848f5
    MPI_Finalize();
Packit 0848f5
    return 0;
Packit 0848f5
}