src/smpi/sample/ring_c.c

   1 /*
   2  * Copyright (c) 2004-2006 The Trustees of Indiana University and Indiana
   3  *                         University Research and Technology
   4  *                         Corporation.  All rights reserved.
   5  * Copyright (c) 2006      Cisco Systems, Inc.  All rights reserved.
   6  *
   7  * Simple ring test program
   8  */
   9
  10 #include <stdio.h>
  11 #include "mpi.h"
  12
  13 int main(int argc, char *argv[])
  14 {
  15     int rank, size, next, prev, message, tag = 201;
  16
  17     /* Start up MPI */
  18
  19     MPI_Init(&argc, &argv);
  20     MPI_Comm_rank(MPI_COMM_WORLD, &rank);
  21     MPI_Comm_size(MPI_COMM_WORLD, &size);
  22
  23     /* Calculate the rank of the next process in the ring.  Use the
  24        modulus operator so that the last process "wraps around" to
  25        rank zero. */
  26
  27     next = (rank + 1) % size;
  28     prev = (rank + size - 1) % size;
  29
  30     /* If we are the "master" process (i.e., MPI_COMM_WORLD rank 0),
  31        put the number of times to go around the ring in the
  32        message. */
  33
  34     if (0 == rank) {
  35         message = 10;
  36
  37         printf("Process 0 sending %d to %d, tag %d (%d processes in ring)\n",
  38                message, next, tag, size);
  39         MPI_Send(&message, 1, MPI_INT, next, tag, MPI_COMM_WORLD);
  40         printf("Process 0 sent to %d\n", next);
  41     }
  42
  43     /* Pass the message around the ring.  The exit mechanism works as
  44        follows: the message (a positive integer) is passed around the
  45        ring.  Each time it passes rank 0, it is decremented.  When
  46        each processes receives a message containing a 0 value, it
  47        passes the message on to the next process and then quits.  By
  48        passing the 0 message first, every process gets the 0 message
  49        and can quit normally. */
  50
  51     sleep(3);
  52
  53     while (1) {
  54         MPI_Recv(&message, 1, MPI_INT, prev, tag, MPI_COMM_WORLD,
  55                  MPI_STATUS_IGNORE);
  56
  57         if (0 == rank) {
  58             --message;
  59             printf("Process 0 decremented value: %d\n", message);
  60         }
  61
  62         MPI_Send(&message, 1, MPI_INT, next, tag, MPI_COMM_WORLD);
  63         if (0 == message) {
  64             printf("Process %d exiting\n", rank);
  65             break;
  66         }
  67     }
  68
  69     /* The last process does one extra send to process 0, which needs
  70        to be received before the program can exit */
  71
  72     if (0 == rank) {
  73         MPI_Recv(&message, 1, MPI_INT, prev, tag, MPI_COMM_WORLD,
  74                  MPI_STATUS_IGNORE);
  75     }
  76
  77     /* All done */
  78
  79     MPI_Finalize();
  80     return 0;
  81 }