← Back to branch summary

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

« back to all changes in this revision

Viewing changes to teshsuite/smpi/mpich-test/pt2pt/hindexed.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:
teshsuite/smpi/mpich-test/pt2pt/hindexed.c
 
1
#include "mpi.h"
 
2
#include <stdio.h>
 
3
/* stdlib.h needed for malloc declaration */
 
4
#include <stdlib.h>
 
5
 
 
6
#if defined(NEEDS_STDLIB_PROTOTYPES)
 
7
#include "protofix.h"
 
8
#endif
 
9
 
 
10
/*
 
11
 * This file tests MPI_Type_hindexed by describing parts of a triangular
 
12
 * matrix, stored in a square matrix, and sending sending it.
 
13
 * 
 
14
 * The matrix is stored in column-major, and the tests use
 
15
 * MPI_Type_vector or MPI_Type_struct to define the elements that are sent
 
16
 */
 
17
 
 
18
int main( int argc, char **argv )
 
19
{
 
20
    MPI_Datatype rowtype, mattype;
 
21
    int          *sbuf, *rbuf;
 
22
    int          rank, mat_n;
 
23
    static int   blens[2] = { 1, 1 };
 
24
    MPI_Datatype types[2] = { MPI_INT, MPI_UB };
 
25
    int          *mat_blens, i ;
 
26
    MPI_Aint     *mat_displs;
 
27
    MPI_Aint     displs[2];
 
28
    MPI_Status   status;
 
29
    int          err, row, col;
 
30
 
 
31
    MPI_Init( &argc, &argv );
 
32
 
 
33
    err = 0;
 
34
    MPI_Comm_rank( MPI_COMM_WORLD, &rank );
 
35
    mat_n = 10;
 
36
    sbuf = (int *) malloc( mat_n * mat_n * sizeof(int) );
 
37
    rbuf = (int *) malloc( mat_n * mat_n * sizeof(int) );
 
38
    if (!sbuf || !rbuf) {
 
39
        MPI_Abort( MPI_COMM_WORLD, 1 );
 
40
    }
 
41
 
 
42
    /* Define a row type based on a strided struct type */
 
43
    displs[0] = 0;
 
44
    displs[1] = mat_n*sizeof(int);
 
45
    MPI_Type_struct( 2, blens, displs, types, &rowtype );
 
46
     MPI_Type_commit( &rowtype ); 
 
47
 
 
48
    /* Define an hindexed type that defines all of the rows of the
 
49
       triangular part of sbuf */
 
50
    
 
51
    mat_blens = (int *)malloc( mat_n * sizeof(int) );
 
52
    mat_displs = (MPI_Aint *)malloc( mat_n * sizeof(MPI_Aint) );
 
53
    for (i=0; i<mat_n; i++) {
 
54
        mat_blens[i] = mat_n - i;
 
55
        MPI_Address( &sbuf[i + i * mat_n], &mat_displs[i] );
 
56
        if (i != 0)
 
57
            mat_displs[i] = mat_displs[i] - mat_displs[0];
 
58
    }
 
59
    mat_displs[0] = 0;
 
60
    MPI_Type_hindexed( mat_n, mat_blens, mat_displs, rowtype, &mattype );
 
61
    MPI_Type_commit( &mattype );
 
62
   // MPI_Type_free( &rowtype );
 
63
 
 
64
    /* Load up the data */
 
65
    for (i=0; i<mat_n * mat_n; i++) {
 
66
        sbuf[i] = i;
 
67
        rbuf[i] = -i;
 
68
    }
 
69
    
 
70
    /* Send it and receive it in the same order */
 
71
    MPI_Sendrecv( sbuf, 1, mattype, rank, 0, rbuf, 1, mattype, rank, 0, 
 
72
                  MPI_COMM_WORLD, &status );
 
73
 
 
74
    for (row = 0; row<mat_n; row++) {
 
75
        for (col = row; col<mat_n; col++) {
 
76
            if (rbuf[row + col*mat_n] != sbuf[row + col*mat_n]) {
 
77
                err++;
 
78
                fprintf( stderr, "mat(%d,%d) = %d, not %d\n",
 
79
                         row, col, rbuf[row+col*mat_n], sbuf[row+col*mat_n] );
 
80
            }
 
81
        }
 
82
    }
 
83
 
 
84
    /* Send hindexed and receive contiguous */
 
85
    MPI_Sendrecv( sbuf, 1, mattype, rank, 1, 
 
86
                  rbuf, (mat_n * (mat_n + 1))/2, MPI_INT, rank, 1, 
 
87
                  MPI_COMM_WORLD, &status );
 
88
    i = 0;
 
89
    for (row = 0; row<mat_n; row++) {
 
90
        for (col = row; col<mat_n; col++) {
 
91
            if (rbuf[i] != sbuf[row + col*mat_n]) {
 
92
                err++;
 
93
                fprintf( stderr, "rbuf(%d,%d) = %d, not %d\n",
 
94
                         row, col, rbuf[i], sbuf[row+col*mat_n] );
 
95
            }
 
96
            i++;
 
97
        }
 
98
    }
 
99
 
 
100
    MPI_Type_free( &mattype );
 
101
    if (err == 0) printf( "Test passed\n" );
 
102
    else          printf( "Test failed with %d errors\n", err );
 
103
 
 
104
    MPI_Finalize();
 
105
    return 0;
 
106
}