1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112
|
/*
* Copyright (C) by Argonne National Laboratory
* See COPYRIGHT in top-level directory
*/
#include "mpi.h"
#include "mpitest.h"
#include <stdlib.h>
#include <stdio.h>
/* The next is for isprint */
#include <ctype.h>
int main(int argc, char *argv[])
{
struct a {
int i;
char c;
} s[10], s1[10];
int j;
int errs = 0;
int rank, size, tsize;
MPI_Aint text, tmp_lb;
int blens[2];
MPI_Aint disps[2];
MPI_Datatype bases[2];
MPI_Datatype str, con;
char *buffer;
int bufsize, position, insize;
MTest_Init(&argc, &argv);
MPI_Comm_rank(MPI_COMM_WORLD, &rank);
MPI_Comm_size(MPI_COMM_WORLD, &size);
for (j = 0; j < 10; j++) {
s[j].i = j + rank;
s[j].c = j + rank + 'a';
}
blens[0] = blens[1] = 1;
disps[0] = 0;
disps[1] = sizeof(int);
bases[0] = MPI_INT;
bases[1] = MPI_CHAR;
MPI_Type_create_struct(2, blens, disps, bases, &str);
MPI_Type_commit(&str);
MPI_Type_contiguous(10, str, &con);
MPI_Type_commit(&con);
MPI_Type_size(con, &tsize);
MPI_Type_get_extent(con, &tmp_lb, &text);
#ifdef DEBUG
printf("Size of MPI array is %d, extent is %d\n", tsize, text);
#endif
#ifdef DEBUG
{
void *p1, *p2;
p1 = s;
p2 = &(s[10].i); /* This statement may fail on some systems */
printf("C array starts at %p and ends at %p for a length of %d\n",
s, &(s[9].c), (char *) p2 - (char *) p1);
}
#endif
MPI_Type_get_extent(str, &tmp_lb, &text);
#ifdef DEBUG
MPI_Type_size(str, &tsize);
printf("Size of MPI struct is %d, extent is %d\n", tsize, (int) text);
printf("Size of C struct is %d\n", sizeof(struct a));
#endif
if (text != sizeof(struct a)) {
printf("Extent of struct a (%d) does not match sizeof (%d)\n",
(int) text, (int) sizeof(struct a));
errs++;
}
MPI_Pack_size(1, con, MPI_COMM_WORLD, &bufsize);
buffer = (char *) malloc(bufsize);
position = 0;
MPI_Pack(s, 1, con, buffer, bufsize, &position, MPI_COMM_WORLD);
insize = position;
position = 0;
MPI_Unpack(buffer, insize, &position, s1, 1, con, MPI_COMM_WORLD);
for (j = 0; j < 10; j++) {
#ifdef DEBUG
printf("%d Sent: %d %c, Got: %d %c\n", rank, s[j].i, s[j].c, s1[j].i, s1[j].c);
#endif
if (s1[j].i != j + rank) {
errs++;
printf("Got s[%d].i = %d (%x); expected %d\n", j, s1[j].i, s1[j].i, j + rank);
}
if (s1[j].c != 'a' + j + rank) {
errs++;
/* If the character is not a printing character,
* this can generate an file that diff, for example,
* believes is a binary file */
if (isprint((int) (s1[j].c))) {
printf("Got s[%d].c = %c; expected %c\n", j, s1[j].c, j + rank + 'a');
} else {
printf("Got s[%d].c = %x; expected %c\n", j, (int) s1[j].c, j + rank + 'a');
}
}
}
free(buffer);
MPI_Type_free(&str);
MPI_Type_free(&con);
MTest_Finalize(errs);
return MTestReturnValue(errs);
}
|