Blame examples/ircpi.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
#include "mpi.h"
Packit 0848f5
#include "stdio.h"
Packit 0848f5
#include <math.h>
Packit 0848f5
Packit 0848f5
/* From Using MPI-2 */
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
    MPI_Win nwin, piwin;
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
Packit 0848f5
    if (myid == 0) {
Packit 0848f5
        MPI_Win_create(&n, sizeof(int), 1, MPI_INFO_NULL,
Packit 0848f5
                       MPI_COMM_WORLD, &nwin);
Packit 0848f5
        MPI_Win_create(&pi, sizeof(double), 1, MPI_INFO_NULL,
Packit 0848f5
                       MPI_COMM_WORLD, &piwin);
Packit 0848f5
    }
Packit 0848f5
    else {
Packit 0848f5
        MPI_Win_create(MPI_BOTTOM, 0, 1, MPI_INFO_NULL,
Packit 0848f5
                       MPI_COMM_WORLD, &nwin);
Packit 0848f5
        MPI_Win_create(MPI_BOTTOM, 0, 1, MPI_INFO_NULL,
Packit 0848f5
                       MPI_COMM_WORLD, &piwin);
Packit 0848f5
    }
Packit 0848f5
    while (1) {
Packit 0848f5
        if (myid == 0) {
Packit 0848f5
            fprintf(stdout, "Enter the number of intervals: (0 quits) ");
Packit 0848f5
            fflush(stdout);
Packit 0848f5
            scanf("%d",&n);
Packit 0848f5
            pi = 0.0;
Packit 0848f5
        }
Packit 0848f5
        MPI_Win_fence(0, nwin);
Packit 0848f5
        if (myid != 0)
Packit 0848f5
            MPI_Get(&n, 1, MPI_INT, 0, 0, 1, MPI_INT, nwin);
Packit 0848f5
        MPI_Win_fence(0, nwin);
Packit 0848f5
        if (n == 0)
Packit 0848f5
            break;
Packit 0848f5
        else {
Packit 0848f5
            h = 1.0 / (double) n;
Packit 0848f5
            sum = 0.0;
Packit 0848f5
            for (i = myid + 1; i <= n; i += numprocs) {
Packit 0848f5
                x = h * ((double)i - 0.5);
Packit 0848f5
                sum += (4.0 / (1.0 + x*x));
Packit 0848f5
            }
Packit 0848f5
            mypi = h * sum;
Packit 0848f5
            MPI_Win_fence( 0, piwin);
Packit 0848f5
            MPI_Accumulate(&mypi, 1, MPI_DOUBLE, 0, 0, 1, MPI_DOUBLE,
Packit 0848f5
                           MPI_SUM, piwin);
Packit 0848f5
            MPI_Win_fence(0, piwin);
Packit 0848f5
            if (myid == 0) {
Packit 0848f5
                fprintf(stdout, "pi is approximately %.16f, Error is %.16f\n",
Packit 0848f5
                        pi, fabs(pi - PI25DT));
Packit 0848f5
                fflush(stdout);
Packit 0848f5
            }
Packit 0848f5
        }
Packit 0848f5
    }
Packit 0848f5
    MPI_Win_free(&nwin);
Packit 0848f5
    MPI_Win_free(&piwin);
Packit 0848f5
    MPI_Finalize();
Packit 0848f5
    return 0;
Packit 0848f5
}