teshsuite/smpi/type-struct/type-struct.c

   1 /* Copyright (c) 2012-2019. 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];
  18     MPI_Datatype type2;
  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 (int j = 0; j < 2; j++)
  61       for (int 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 }