haskell-mpi-1.0.0: test/examples/HaskellAndC/Rank0.c

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

#define MAX_MSG 100

int main(int argc, char **argv)
{
   int rank, size, i;
   int msg[MAX_MSG];
   MPI_Status status;

   MPI_Init(NULL, NULL);
   MPI_Comm_rank(MPI_COMM_WORLD, &rank);
   MPI_Comm_size(MPI_COMM_WORLD, &size);

   printf("C process with rank %d world with size %d\n", rank, size);

   if (rank == 0)
   {
      for (i = 0; i < MAX_MSG; i++)
      {
         msg[i] = i+1;
      }
      MPI_Send(msg, MAX_MSG, MPI_INT, 1, 0, MPI_COMM_WORLD);
      MPI_Recv(msg, MAX_MSG, MPI_INT, 1, 0, MPI_COMM_WORLD, &status);

      for (i = 0; i < MAX_MSG; i++)
      {
         printf("%d ", msg[i]);
      }
   }
   else
   {
      printf ("This program must be rank 0\n");
   }
   MPI_Finalize();

   return 0;
}