Subversion Repositories programming

#include <stdio.h>
#include <limits.h>
#include <mpi.h>

const int ROOT_NODE = 0;
int numprocs;
int myid;

/**
 * NOTE: Accessing "matrices" in this program is a little strange.
 * I decided to implement matrices using an array. To get to a[row][col],
 * you would use a[row*a_size+col]. Pretty simple actually.
 */

int* allocate (int rows, int cols)
{
    int *temp = NULL;

    if ((temp = (int*) malloc (rows * cols * sizeof(int))) == NULL)
        printf ("Could not allocate memory for array of size: %d x %d\n", rows, cols);

    return temp;
}

void print_matrix (int *b, int n, const char *s)
{
    int i,j;

    if (n<=10) // n<=10
    {
        printf ("%s\n", s);

        for (i=0; i<n; i++)
        {
            for (j=0; j<n; j++)
                printf ("%d\t", b[i*n+j]);

            printf ("\n");
        }
        printf ("\n");
    }
    else if (n<=100) // 10<n<=100
    {
        printf ("%s\n", s);

        // First 2 rows
        for (i=0; i<2; i++)
        {
            for (j=0; j<n; j++)
                printf ("%d\t", b[i*n+j]);

            printf ("\n");
        }
        printf ("\n");

        // First 2 cols
        for (i=0; i<n; i++)
        {
            for (j=0; j<2; j++)
                printf ("%d\t", b[i*n+j]);

            printf ("\n");
        }
    }
}

void matrix_mult (int *b, int *c, int n)
{
    int i,j,k,ii,temp;

    /* Multiply our slice of the matrix, store it in c */
    for (ii=0; ii<n/numprocs; ii++)
    {
        i = ii + myid * n/numprocs; // i is the offset into b

        for (j=0; j<n; j++)
        {
            // MATRIX_MULT
            // c[i*n+j]=0;

            // SHORTEST PATH
            c[ii*n+j] = INT_MAX;

            for (k=0; k<n; k++)
            {
                // MATRIX MULT
                // c[i*n+j] += a[i*n+k] * b[k*n+j];

                // SHORTEST PATH
                if (b[i*n+k] == INT_MAX || b[k*n+j] == INT_MAX)
                    temp = INT_MAX;
                else
                    temp = b[i*n+k] + b[k*n+j];

                if (temp < c[ii*n+j])
                    c[ii*n+j] = temp;
            }
        }
    }
}

int main (int argc, char *argv[])
{
    const int n = 4;
    int *b, *c;
    int i, j, k, temp, t;
    MPI_Status status;

    /* Get all of the necessary info for each process */
    MPI_Init (&argc, &argv);
    MPI_Comm_rank (MPI_COMM_WORLD, &myid);
    MPI_Comm_size (MPI_COMM_WORLD, &numprocs);

    /* Do a quick sanity check for the correct number of processors */
    if (n % numprocs != 0)
    {
        if (myid == 0)
        {
            printf ("%d mod %d must be zero!\n", n, numprocs);
            printf ("You should probably change the -np parameter\n");
        }

        MPI_Finalize ();
        return 1;
    }

    /* Allocate all of b for each process, since we'll need it */
    b = allocate (n, n);

    if (myid == ROOT_NODE)
    {
        /* Allocate the full c, since we need to initialize it */
        c = allocate (n, n);

        /* Initialize values */
        for (i=0; i<n; i++)
            for (j=0; j<n; j++)
                b[i*n+j] = INT_MAX;

        b[0] = 0;
        b[1] = 2;
        b[2] = INT_MAX;
        b[3] = 10;

        b[4] = 2;
        b[5] = 0;
        b[6] = 2;
        b[7] = INT_MAX;

        b[8]  = INT_MAX;
        b[9]  = 2;
        b[10] = 0;
        b[11] = 3;

        b[12] = 10;
        b[13] = INT_MAX;
        b[14] = 3;
        b[15] = 0;

        if (n<=100)
            print_matrix (b, n, "Input Matrix:");
        else
        {
            printf ("Beginning calculation for all shortest paths\n");
            printf ("for a graph with %d nodes\n", n);
        }

    }
    else
    {
        /* Allocate our slice of c, since that's all we need */
        c = allocate (n/numprocs, n);
    }

    /* Send all of b to each process, since they need all of it */
    MPI_Bcast (b, n*n, MPI_INT, ROOT_NODE, MPI_COMM_WORLD);

    /* Run our piece of the calculation.
     * Synchronize data with all other processes at the end of each step. */
    for (t=0; t<ceil(log(n-1)); t++)
    {
        matrix_mult (b, c, n);
        MPI_Allgather (c, n*n/numprocs, MPI_INT, b, n*n/numprocs, MPI_INT, MPI_COMM_WORLD);
    }

    /* Print out the result matrix. */
    if (myid == ROOT_NODE)
    {
        if (n<=100)
            print_matrix (b, n, "Result Matrix:");
        else
        {
            printf ("Finished calculation for all shortest paths\n");
            printf ("for a graph with %d nodes\n", n);
        }
    }

    /* Free all of the memory we allocated earlier */
    free (b);
    free (c);

    MPI_Finalize ();
    return 0;
}