Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
Kill the comments intending to explain which part of the autotools were converted...
[simgrid.git] / examples / smpi / bcbench.c
1 /* Copyright (c) 2009, 2010. The SimGrid Team.
2  * All rights reserved.                                                     */
3
4 /* This program is free software; you can redistribute it and/or modify it
5  * under the terms of the license (GNU LGPL) which comes with this package. */
6
7 #include <stdio.h>
8 #include <stdlib.h>
9 #include <mpi.h>
10
11 #define GETTIMEOFDAY_ERROR 1
12
13 #define N_START 1
14 #define N_STOP  1024*1024
15 #define N_NEXT  (N*2)
16 #define ITER    100
17 #define ONE_MILLION 1000000.0
18 #define RAND_SEED 842270
19
20 int main(int argc, char *argv[])
21 {
22
23   int size, rank;
24   int N;
25   struct timeval *start_time = NULL, *stop_time = NULL;
26   double seconds = 0;
27   int i, j;
28   char *buffer;
29   int check;
30
31   srandom(RAND_SEED);
32
33   MPI_Init(&argc, &argv);
34
35   MPI_Comm_rank(MPI_COMM_WORLD, &rank);
36   MPI_Comm_size(MPI_COMM_WORLD, &size);
37
38   if (0 == rank) {
39     start_time = (struct timeval *) malloc(sizeof(struct timeval));
40     stop_time = (struct timeval *) malloc(sizeof(struct timeval));
41   }
42
43   for (N = N_START; N <= N_STOP; N = N_NEXT) {
44
45     buffer = malloc(sizeof(char) * N);
46
47     if (0 == rank) {
48       for (j = 0; j < N; j++) {
49         buffer[j] = (char) (random() % 256);
50       }
51       if (-1 == gettimeofday(start_time, NULL)) {
52         printf("couldn't set start_time on node 0!\n");
53         MPI_Abort(MPI_COMM_WORLD, GETTIMEOFDAY_ERROR);
54         exit(EXIT_FAILURE);
55       }
56     }
57
58     for (i = 0; i < ITER; i++) {
59       MPI_Bcast(buffer, N, MPI_BYTE, 0, MPI_COMM_WORLD);
60       if (0 == rank) {
61         for (j = 1; j < size; j++) {
62           MPI_Recv(&check, 1, MPI_INT, MPI_ANY_SOURCE, 1, MPI_COMM_WORLD,
63                    MPI_STATUS_IGNORE);
64         }
65       } else {
66         MPI_Send(&rank, 1, MPI_INT, 0, 1, MPI_COMM_WORLD);
67       }
68     }
69
70     if (0 == rank) {
71       if (-1 == gettimeofday(stop_time, NULL)) {
72         printf("couldn't set start_time on node 0!\n");
73         MPI_Abort(MPI_COMM_WORLD, GETTIMEOFDAY_ERROR);
74         exit(EXIT_FAILURE);
75       }
76       seconds =
77           (double) (stop_time->tv_sec - start_time->tv_sec) +
78           (double) (stop_time->tv_usec -
79                     start_time->tv_usec) / ONE_MILLION;
80     }
81
82     free(buffer);
83
84     if (0 == rank) {
85       printf("N: %10d, iter: %d, time: %10f s, avg rate: %12f Mbps\n", N,
86              ITER, seconds,
87              ((double) N * ITER * 8) / (1024.0 * 1024.0 * seconds));
88     }
89
90   }
91
92   if (0 == rank) {
93     free(start_time);
94     free(stop_time);
95   }
96
97   MPI_Finalize();
98
99   return 0;
100 }