teshsuite/smpi/mpich3-test/datatype/vecblklen.c

   1 /* -*- Mode: C; c-basic-offset:4 ; indent-tabs-mode:nil ; -*- */
   2 /*
   3  *  (C) 2014 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 <string.h>
  10 #include "mpitest.h"
  11
  12 /* Inspired by the Intel MPI_Type_vector_blklen test.
  13    Added to include a test of a dataloop optimization that failed.
  14 */
  15 int main( int argc, char *argv[] )
  16 {
  17     MPI_Datatype ot, ot2, newtype;
  18     int position, psize, insize, outsize;
  19     signed char *inbuf=0, *outbuf=0, *pbuf=0, *p;
  20     int  i, j, k;
  21     int  errs = 0;
  22     int  veccount=16, stride=16;
  23
  24     MTest_Init( &argc, &argv );
  25     /*
  26      * Create a type with some padding
  27      */
  28     MPI_Type_contiguous( 59, MPI_CHAR, &ot );
  29     MPI_Type_create_resized( ot, 0, 64, &ot2 );
  30     /*
  31       Use a vector type with a block size equal to the stride - thus
  32       tiling the target memory with copies of old type.  This is not
  33       a contiguous copy since oldtype has a gap at the end.
  34     */
  35     MPI_Type_vector( veccount, stride, stride, ot2, &newtype );
  36     MPI_Type_commit( &newtype );
  37
  38     insize = veccount * stride * 64;
  39     outsize = insize;
  40     inbuf = (char *)malloc( insize );
  41     outbuf = (char *)malloc( outsize );
  42     for (i=0; i<outsize; i++) {
  43         inbuf[i] = i % 64;
  44         outbuf[i] = -1;
  45     }
  46
  47     MPI_Pack_size( 1, newtype, MPI_COMM_WORLD, &psize );
  48     pbuf = (char *)malloc( psize );
  49
  50     position = 0;
  51     MPI_Pack( inbuf, 1, newtype, pbuf, psize, &position, MPI_COMM_WORLD );
  52     psize    = position;
  53     position = 0;
  54     MPI_Unpack( pbuf, psize, &position, outbuf, 1, newtype, MPI_COMM_WORLD );
  55
  56
  57     /* Check the output */
  58     p = outbuf;
  59     for (i=0; i<veccount; i++) {
  60         for (j=0; j<stride; j++) {
  61             for (k=0; k<59; k++) {
  62                 if (*p != k % 64) {
  63                     errs++;
  64                     fprintf( stderr, "[%d,%d,%d]expected %d but saw %d\n",
  65                              i, j, k, (k%64), *p );
  66                 }
  67                 p++;
  68             }
  69             for (k=59; k<64; k++) {
  70                 if (*p != -1) {
  71                     errs++;
  72                     fprintf( stderr, "[%d,%d,%d]expected -1 but saw %d\n",
  73                              i, j, k, *p );
  74                 }
  75                 p++;
  76             }
  77         }
  78     }
  79
  80     free( pbuf );
  81     free( inbuf );
  82     free( outbuf );
  83
  84     MPI_Type_free( &ot );
  85     MPI_Type_free( &ot2 );
  86     MPI_Type_free( &newtype );
  87     MTest_Finalize( errs );
  88     MPI_Finalize();
  89
  90     return 0;
  91 }