teshsuite/smpi/mpich3-test/datatype/simple-resized.c

   1 /* -*- Mode: C; c-basic-offset:4 ; indent-tabs-mode:nil ; -*- */
   2 /*
   3  *  (C) 2001 by Argonne National Laboratory.
   4  *      See COPYRIGHT in top-level directory.
   5  */
   6 #include "mpi.h"
   7 #include <stdio.h>
   8 #include <stdlib.h>
   9 #include "mpitestconf.h"
  10 #ifdef HAVE_STRING_H
  11 #include <string.h>
  12 #endif
  13
  14 static int verbose = 0;
  15
  16 /* tests */
  17 int derived_resized_test(void);
  18
  19 /* helper functions */
  20 int parse_args(int argc, char **argv);
  21
  22 int main(int argc, char **argv)
  23 {
  24     int err, errs = 0;
  25
  26     MPI_Init(&argc, &argv); /* MPI-1.2 doesn't allow for MPI_Init(0,0) */
  27     parse_args(argc, argv);
  28
  29     /* To improve reporting of problems about operations, we
  30        change the error handler to errors return */
  31     MPI_Comm_set_errhandler( MPI_COMM_WORLD, MPI_ERRORS_RETURN );
  32
  33     /* perform some tests */
  34     err = derived_resized_test();
  35     if (err && verbose) fprintf(stderr, "%d errors in derived_resized test.\n",
  36                                 err);
  37     errs += err;
  38
  39     /* print message and exit */
  40     if (errs) {
  41         fprintf(stderr, "Found %d errors\n", errs);
  42     }
  43     else {
  44         printf(" No Errors\n");
  45     }
  46     MPI_Finalize();
  47     return 0;
  48 }
  49
  50 /* derived_resized_test()
  51  *
  52  * Tests behavior with resizing of a simple derived type.
  53  *
  54  * Returns the number of errors encountered.
  55  */
  56 int derived_resized_test(void)
  57 {
  58     int err, errs = 0;
  59
  60     int count = 2;
  61     MPI_Datatype newtype, resizedtype;
  62
  63     int size;
  64     MPI_Aint extent;
  65
  66     err = MPI_Type_contiguous(count,
  67                              MPI_INT,
  68                              &newtype);
  69     if (err != MPI_SUCCESS) {
  70         if (verbose) {
  71             fprintf(stderr,
  72                     "error creating type in derived_resized_test()\n");
  73         }
  74         errs++;
  75     }
  76
  77     err = MPI_Type_create_resized(newtype,
  78                                   (MPI_Aint) 0,
  79                                   (MPI_Aint) (2*sizeof(int) + 10),
  80                                   &resizedtype);
  81
  82     err = MPI_Type_size(resizedtype, &size);
  83     if (err != MPI_SUCCESS) {
  84         if (verbose) {
  85             fprintf(stderr,
  86                     "error obtaining type size in derived_resized_test()\n");
  87         }
  88         errs++;
  89     }
  90
  91     if (size != 2*sizeof(int)) {
  92         if (verbose) {
  93             fprintf(stderr,
  94                     "error: size != %d in derived_resized_test()\n", (int) (2*sizeof(int)));
  95         }
  96         errs++;
  97     }
  98
  99     err = MPI_Type_extent(resizedtype, &extent);
 100     if (err != MPI_SUCCESS) {
 101         if (verbose) {
 102             fprintf(stderr,
 103                     "error obtaining type extent in derived_resized_test()\n");
 104         }
 105         errs++;
 106     }
 107
 108     if (extent != 2*sizeof(int) + 10) {
 109         if (verbose) {
 110             fprintf(stderr,
 111                     "error: invalid extent (%d) in derived_resized_test(); should be %d\n",
 112                     (int) extent,
 113                     (int) (2*sizeof(int) + 10));
 114         }
 115         errs++;
 116     }
 117
 118     MPI_Type_free( &newtype );
 119     MPI_Type_free( &resizedtype );
 120
 121     return errs;
 122 }
 123
 124
 125 int parse_args(int argc, char **argv)
 126 {
 127     /*
 128     int ret;
 129
 130     while ((ret = getopt(argc, argv, "v")) >= 0)
 131     {
 132         switch (ret) {
 133             case 'v':
 134                 verbose = 1;
 135                 break;
 136         }
 137     }
 138     */
 139     if (argc > 1 && strcmp(argv[1], "-v") == 0)
 140         verbose = 1;
 141     return 0;
 142 }
 143