← Back to branch summary

~ubuntu-branches/ubuntu/raring/simgrid/raring

« back to all changes in this revision

Viewing changes to examples/smpi/sendtest.c

  • Committer: Package Import Robot
  • Author(s): Martin Quinson
  • Date: 2013-01-31 00:24:51 UTC
  • mfrom: (10.1.6 sid)
  • Revision ID: package-import@ubuntu.com-20130131002451-krejhf7w7h24lpsc
Tags: 3.9~rc1-1
* New upstream release: the "Grasgory" release. Major changes:
  - Gras was completely removed from this version.
  - Documentation reorganization to ease browsing it.
  - New default value for the TCP_gamma parameter: 4MiB

Show diffs side-by-side

added added

removed removed

Lines of Context:
examples/smpi/sendtest.c
1
 
/* A simple example pingpong pogram to test MPI_Send and MPI_Recv */
2
 
 
3
 
/* Copyright (c) 2009, 2010. The SimGrid Team.
4
 
 * All rights reserved.                                                     */
5
 
 
6
 
/* This program is free software; you can redistribute it and/or modify it
7
 
 * under the terms of the license (GNU LGPL) which comes with this package. */
8
 
 
9
 
#include <stdio.h>
10
 
#include <stdlib.h>
11
 
#include <sys/time.h>
12
 
#include <mpi.h>
13
 
 
14
 
int main(int argc, char *argv[])
15
 
{
16
 
    int rank;
17
 
    int size;
18
 
    MPI_Status status;
19
 
 
20
 
    int n = 0, m = 0, bytes = 0, workusecs = 0;
21
 
    int currusecs;
22
 
  
23
 
    char *buf = NULL;
24
 
    int i, j;
25
 
    struct timeval start_time, stop_time;
26
 
    struct timeval start_pause, curr_pause;
27
 
    unsigned long usecs;
28
 
 
29
 
    MPI_Init(&argc, &argv); /* Initialize MPI */
30
 
    MPI_Comm_size(MPI_COMM_WORLD, &size);   /* Get nr of tasks */
31
 
    MPI_Comm_rank(MPI_COMM_WORLD, &rank);   /* Get id of this process */
32
 
 
33
 
    if (size != 2) {
34
 
        printf("run this program with exactly 2 processes (-np 2)\n");
35
 
        MPI_Finalize();
36
 
        exit(0);
37
 
    }
38
 
 
39
 
    if (0 == rank) {
40
 
        if (argc > 1 && isdigit(argv[1][0])) {
41
 
            n = atoi(argv[1]);
42
 
        }
43
 
        if (argc > 2 && isdigit(argv[2][0])) {
44
 
            m = atoi(argv[2]);
45
 
        }
46
 
        if (argc > 3 && isdigit(argv[3][0])) {
47
 
            bytes = atoi(argv[3]);
48
 
        }
49
 
        if (argc > 4 && isdigit(argv[4][0])) {
50
 
            workusecs = atoi(argv[4]);
51
 
        }
52
 
        buf = malloc(sizeof(char) * bytes);
53
 
        for (i = 0; i < bytes; i++) buf[i] = i % 256;
54
 
        MPI_Send(&n, 1, MPI_INT, 1, 0, MPI_COMM_WORLD);
55
 
        MPI_Send(&m, 1, MPI_INT, 1, 0, MPI_COMM_WORLD);
56
 
        MPI_Send(&bytes, 1, MPI_INT, 1, 0, MPI_COMM_WORLD);
57
 
        MPI_Send(&workusecs, 1, MPI_INT, 1, 0, MPI_COMM_WORLD);
58
 
        MPI_Barrier(MPI_COMM_WORLD);
59
 
        gettimeofday(&start_time, NULL);
60
 
        for (i = 0; i < n; i++) {
61
 
            for (j = 0; j < m; j++) {
62
 
                MPI_Send(buf, bytes, MPI_CHAR, 1, 0, MPI_COMM_WORLD);
63
 
                gettimeofday(&start_pause, NULL);
64
 
                currusecs = 0;
65
 
                while (currusecs < workusecs) {
66
 
                    gettimeofday(&curr_pause, NULL);
67
 
                    currusecs = (curr_pause.tv_sec - start_pause.tv_sec) * 1e6 + curr_pause.tv_usec - start_pause.tv_usec;
68
 
                }
69
 
            }
70
 
            MPI_Recv(buf, bytes, MPI_CHAR, 1, 0, MPI_COMM_WORLD, &status);
71
 
        }
72
 
        gettimeofday(&stop_time, NULL);
73
 
        usecs = (stop_time.tv_sec - start_time.tv_sec) * 1e6 + stop_time.tv_usec - start_time.tv_usec;
74
 
        printf("n: %d m: %d bytes: %d sleep: %d usecs: %u\n", n, m, bytes, workusecs, usecs);
75
 
    } else {
76
 
        MPI_Recv(&n, 1, MPI_INT, 0, 0, MPI_COMM_WORLD, &status);
77
 
        MPI_Recv(&m, 1, MPI_INT, 0, 0, MPI_COMM_WORLD, &status);
78
 
        MPI_Recv(&bytes, 1, MPI_INT, 0, 0, MPI_COMM_WORLD, &status);
79
 
        MPI_Recv(&workusecs, 1, MPI_INT, 0, 0, MPI_COMM_WORLD, &status);
80
 
        buf = malloc(sizeof(char) * bytes);
81
 
        MPI_Barrier(MPI_COMM_WORLD);
82
 
        for (i = 0; i < n; i++) {
83
 
            for (j = 0; j < m; j++) {
84
 
                MPI_Recv(buf, bytes, MPI_CHAR, 0, 0, MPI_COMM_WORLD, &status);
85
 
            }
86
 
            MPI_Send(buf, bytes, MPI_CHAR, 0, 0, MPI_COMM_WORLD);
87
 
        }
88
 
    }
89
 
    free(buf);
90
 
    MPI_Finalize();
91
 
    return 0;
92
 
}