Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
835b852413e47a12e97c1d0b57aee3534371f10c
[simgrid.git] / teshsuite / smpi / type-struct / type-struct.c
1 /* Copyright (c) 2012-2014. 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 "mpi.h"
9
10 int main(int argc, char **argv)
11 {
12     int          rank;
13     struct { int a;int c; double b;int tab[2][3];} value;
14     MPI_Datatype mystruct;
15     int          blocklens[3];
16     MPI_Aint     indices[3];
17     MPI_Datatype old_types[3], type2;
18     int i,j;
19
20     MPI_Init( &argc, &argv );
21
22     MPI_Comm_rank( MPI_COMM_WORLD, &rank );
23
24     int tab[2][3]={{1*rank,2*rank,3*rank},{7*rank,8*rank,9*rank}}; 
25     MPI_Type_contiguous(3, MPI_INT, &type2);
26     MPI_Type_commit(&type2);
27
28     /* One value of each type, and two for the contiguous one */
29     blocklens[0] = 1;
30     blocklens[1] = 1;
31     blocklens[2] = 2;
32     /* The base types */
33     old_types[0] = MPI_INT;
34     old_types[1] = MPI_DOUBLE;
35     old_types[2] = type2;
36     /* The locations of each element */
37     MPI_Address( &value.a, &indices[0] );
38     MPI_Address( &value.b, &indices[1] );
39     MPI_Address( &tab, &indices[2] );
40     /* Make relative */
41     indices[2] = indices[2] - indices[0];
42     indices[1] = indices[1] - indices[0];
43     indices[0] = 0;
44
45     MPI_Type_struct( 3, blocklens, indices, old_types, &mystruct );
46     MPI_Type_commit( &mystruct );
47
48     if (rank == 0){
49       value.a=-2;
50       value.b=8.0;
51     }else{
52       value.a=10000;
53       value.b=5.0; 
54     }
55
56     MPI_Bcast( &value, 1, mystruct, 0, MPI_COMM_WORLD );
57
58     printf( "Process %d got %d (-2?) and %f (8.0?), tab (should be all 0): ", rank, value.a, value.b );
59
60     for(j=0; j<2;j++ )
61       for(i=0; i<3;i++ )
62         printf("%d ", tab[j][i]);
63     printf("\n");
64
65     /* Clean up the type */
66     MPI_Type_free( &mystruct );
67     MPI_Type_free( &type2 );
68     MPI_Finalize( );
69     return 0;
70 }