sendmany.c

来自「MPICH是MPI的重要研究,提供了一系列的接口函数,为并行计算的实现提供了编程」· C语言 代码 · 共 85 行

#include <stdio.h>
#ifdef HAVE_UNISTD_H
#include <unistd.h>
#endif
#include <stdlib.h>
#include <assert.h>
#include "mpi.h"

#if defined(NEEDS_STDLIB_PROTOTYPES)
#include "protofix.h"
#endif

#define MAXPES 32
#define MYBUFSIZE 16*1024
static int buffer[MAXPES][MYBUFSIZE];

#define NUM_RUNS 10


int main ( int argc, char *argv[] )
{
  int i;
  int count, size;
  int self, npes;
  double secs;
  MPI_Request request[MAXPES];
  MPI_Status status;


  MPI_Init (&argc, &argv);
  MPI_Comm_rank (MPI_COMM_WORLD, &self);
  MPI_Comm_size (MPI_COMM_WORLD, &npes);

  assert (npes <= MAXPES);

  for (size = 1; size  <= MYBUFSIZE ; size += size)
    {

      secs = -MPI_Wtime ();
      for (count = 0; count < NUM_RUNS; count++)
	{
	  MPI_Barrier (MPI_COMM_WORLD);

	  for (i = 0; i < npes; i++)
	    {
	      if (i == self)
		continue;
	      MPI_Irecv (buffer[i], size, MPI_INT, i,
			 MPI_ANY_TAG, MPI_COMM_WORLD, &request[i]);
	    }

	  for (i = 0; i < npes; i++)
	    {
	      if (i == self)
		continue;
	      MPI_Send (buffer[self], size, MPI_INT, i, 0, MPI_COMM_WORLD);
	    }

	  for (i = 0; i < npes; i++)
	    {
	      if (i == self)
		continue;
	      MPI_Wait (&request[i], &status);
	    }

	}
      MPI_Barrier (MPI_COMM_WORLD);
      secs += MPI_Wtime ();

      if (self == 0)
	{
	  secs = secs / (double) NUM_RUNS;
	  printf ( "length = %d ints\n", size );
	  fflush(stdout);
/*
	  printf ("%f\n",
		  (double) (size * sizeof (int) * (npes - 1)) /
		    (secs * 1024.0 * 1024.0));
 */
	}
    }
  MPI_Finalize();
  return (0);
}