PRACE NPT

#include <stdio.h>

#include <stdlib.h>

#include "mpi.h"

#include "fundefs.h"

#define  MAXNOD 1000000

int me, nodes;

int offset[MAXNOD][2], sizes[MAXNOD][2];

int main( int argc, char* argv[] )

{ 

   MPI_Comm comm = MPI_COMM_WORLD;

   int      ncols, nrows, nelmts;

   int      lrows, lelmts;

   int      i, j, k;

   int      *indx, *rowp;

   int      ok, gok;

   double   *matvals, *invec, *outvec;

   double   fill, mflops, gmflops, time, gtime;

   FILE     *inl;

// ------------------------------------------------------------------------

   MPI_Init( &argc, &argv );

   MPI_Comm_rank( comm, &me );

   MPI_Comm_size( comm, &nodes );

   if ( me == 0 ) {

      state( "mod2as" );

      prthead( nodes );

   }

   inl = fopen( "mod2as.in", "r" );

   while( ( fscanf( inl, "%d%d%d\n", &ncols, &nrows, &nelmts ) != EOF ) ) {

      if ( nrows < nodes ) {

         if ( me == 0 ) {

            printf( "No. of rows < No. of processes: Increase problem size.\n");

         }

         MPI_Finalize();

         exit;

      }

// ------------------------------------------------------------------------

// --- Partition problem row-wise and allocate.

      lrows   = part( nrows, me, nodes );

      lelmts  = part( nelmts, me, nodes );         

      indx    = calloc( lelmts, sizeof( int ) );

      rowp    = calloc( lrows, sizeof ( int ) );

      matvals = calloc( lelmts, sizeof ( double ) );

      invec   = calloc( ncols, sizeof ( double ) );

      outvec  = calloc( lrows, sizeof ( double ) );

      getmatvec( ncols, lrows, lelmts, indx, rowp, matvals, invec );

      time = MPI_Wtime();

      spmxv( lrows, lelmts, indx, rowp, matvals, invec, outvec );

      time = MPI_Wtime() - time;

      ok = check( ncols, lrows, lelmts, indx, rowp, outvec );

      mflops = 2.0*1.0e-6*(double)nelmts/time;

      MPI_Reduce( &ok, &gok, 1, MPI_INT, MPI_LAND, 0, comm );

      MPI_Reduce( &mflops, &gmflops, 1, MPI_DOUBLE, MPI_MIN, 0, comm );

      MPI_Reduce( &time, &gtime, 1, MPI_DOUBLE, MPI_MAX, 0, comm );

      fill   = 1.0e2*(double)nelmts/((double)(ncols*nrows));

      if ( me == 0 ) prtspeed( ncols, nrows, fill, gtime, gmflops, gok ); 

      free( indx ); free( rowp ); free( matvals ); free( invec );

      free( outvec );

   }

   if ( me == 0 ) {

      printf( "-------------------------------------------------------\n" );

   }

   MPI_Finalize();

}