--- /dev/null
+/* -*- Mode: C; c-basic-offset:4 ; indent-tabs-mode:nil ; -*- */
+/*
+ *
+ * (C) 2014 by Argonne National Laboratory.
+ * See COPYRIGHT in top-level directory.
+ */
+#include "mtest_datatype.h"
+#if defined(HAVE_STDIO_H) || defined(STDC_HEADERS)
+#include <stdio.h>
+#endif
+#if defined(HAVE_STDLIB_H) || defined(STDC_HEADERS)
+#include <stdlib.h>
+#endif
+#if defined(HAVE_STRING_H) || defined(STDC_HEADERS)
+#include <string.h>
+#endif
+#ifdef HAVE_STDARG_H
+#include <stdarg.h>
+#endif
+/* The following two includes permit the collection of resource usage
+ data in the tests
+ */
+#ifdef HAVE_SYS_TIME_H
+#include <sys/time.h>
+#endif
+#ifdef HAVE_SYS_RESOURCE_H
+#include <sys/resource.h>
+#endif
+#include <errno.h>
+
+
+/* ------------------------------------------------------------------------ */
+/* General datatype routines */
+/* ------------------------------------------------------------------------ */
+
+static void *MTestTypeFree(MTestDatatype * mtype)
+{
+ if (mtype->buf)
+ free(mtype->buf);
+ if (mtype->displs)
+ free(mtype->displs);
+ if (mtype->displ_in_bytes)
+ free(mtype->displ_in_bytes);
+ if (mtype->index)
+ free(mtype->index);
+ if (mtype->old_datatypes)
+ free(mtype->old_datatypes);
+ mtype->buf = NULL;
+ mtype->displs = NULL;
+ mtype->displ_in_bytes = NULL;
+ mtype->index = NULL;
+ mtype->old_datatypes = NULL;
+
+ return 0;
+}
+
+static inline void MTestTypeReset(MTestDatatype * mtype)
+{
+ mtype->isBasic = 0;
+ mtype->printErrors = 0;
+ mtype->buf = NULL;
+
+ mtype->datatype = MPI_DATATYPE_NULL;
+ mtype->count = 0;
+ mtype->nblock = 0;
+ mtype->index = NULL;
+
+ mtype->stride = 0;
+ mtype->blksize = 0;
+ mtype->displ_in_bytes = NULL;
+
+ mtype->displs = NULL;
+ mtype->basesize = 0;
+
+ mtype->old_datatypes = NULL;
+
+ mtype->arr_sizes[0] = 0;
+ mtype->arr_sizes[1] = 0;
+ mtype->arr_subsizes[0] = 0;
+ mtype->arr_subsizes[1] = 0;
+ mtype->arr_starts[0] = 0;
+ mtype->arr_starts[1] = 0;
+ mtype->order = 0;
+
+ mtype->InitBuf = NULL;
+ mtype->FreeBuf = NULL;
+ mtype->CheckBuf = NULL;
+}
+
+/* ------------------------------------------------------------------------ */
+/* Datatype routines for contiguous datatypes */
+/* ------------------------------------------------------------------------ */
+/*
+ * Initialize buffer of basic datatype
+ */
+static void *MTestTypeContigInit(MTestDatatype * mtype)
+{
+ MPI_Aint extent = 0, lb = 0, size;
+ int merr;
+
+ if (mtype->count > 0) {
+ unsigned char *p;
+ MPI_Aint i, totsize;
+ merr = MPI_Type_get_extent(mtype->datatype, &lb, &extent);
+ if (merr)
+ MTestPrintError(merr);
+
+ size = extent + lb;
+ totsize = size * mtype->count;
+ if (!mtype->buf) {
+ mtype->buf = (void *) malloc(totsize);
+ }
+ p = (unsigned char *) (mtype->buf);
+ if (!p) {
+ char errmsg[128] = { 0 };
+ sprintf(errmsg, "Out of memory in %s", __FUNCTION__);
+ MTestError(errmsg);
+ }
+ for (i = 0; i < totsize; i++) {
+ p[i] = (unsigned char) (0xff ^ (i & 0xff));
+ }
+ }
+ else {
+ if (mtype->buf) {
+ free(mtype->buf);
+ }
+ mtype->buf = 0;
+ }
+ return mtype->buf;
+}
+
+/*
+ * Check value of received basic datatype buffer.
+ */
+static int MTestTypeContigCheckbuf(MTestDatatype * mtype)
+{
+ unsigned char *p;
+ unsigned char expected;
+ int err = 0, merr;
+ MPI_Aint i, totsize, size, extent = 0, lb = 0;
+
+ p = (unsigned char *) mtype->buf;
+ if (p) {
+ merr = MPI_Type_get_extent(mtype->datatype, &lb, &extent);
+ if (merr)
+ MTestPrintError(merr);
+
+ size = lb + extent;
+ totsize = size * mtype->count;
+ for (i = 0; i < totsize; i++) {
+ expected = (unsigned char) (0xff ^ (i & 0xff));
+ if (p[i] != expected) {
+ err++;
+ if (mtype->printErrors && err < 10) {
+ printf("Data expected = %x but got p[%ld] = %x\n", expected, i, p[i]);
+ fflush(stdout);
+ }
+ }
+ }
+ }
+ return err;
+}
+
+
+/* ------------------------------------------------------------------------ */
+/* Datatype routines for vector datatypes */
+/* ------------------------------------------------------------------------ */
+
+/*
+ * Initialize buffer of vector datatype
+ */
+static void *MTestTypeVectorInit(MTestDatatype * mtype)
+{
+ MPI_Aint extent = 0, lb = 0, size, totsize, dt_offset, byte_offset;
+ int merr;
+
+ if (mtype->count > 0) {
+ unsigned char *p;
+ MPI_Aint k, j;
+ int i, nc;
+
+ merr = MPI_Type_get_extent(mtype->datatype, &lb, &extent);
+ if (merr)
+ MTestPrintError(merr);
+
+ size = extent + lb;
+ totsize = mtype->count * size;
+ if (!mtype->buf) {
+ mtype->buf = (void *) malloc(totsize);
+ }
+ p = (unsigned char *) (mtype->buf);
+ if (!p) {
+ char errmsg[128] = { 0 };
+ sprintf(errmsg, "Out of memory in %s", __FUNCTION__);
+ MTestError(errmsg);
+ }
+
+ /* First, set to -1 */
+ for (k = 0; k < totsize; k++)
+ p[k] = 0xff;
+
+ /* Now, set the actual elements to the successive values.
+ * We require that the base type is a contiguous type */
+ nc = 0;
+ dt_offset = 0;
+ /* For each datatype */
+ for (k = 0; k < mtype->count; k++) {
+ /* For each block */
+ for (i = 0; i < mtype->nblock; i++) {
+ byte_offset = dt_offset + i * mtype->stride;
+ /* For each byte */
+ for (j = 0; j < mtype->blksize; j++) {
+ p[byte_offset + j] = (unsigned char) (0xff ^ (nc & 0xff));
+ nc++;
+ }
+ }
+ dt_offset += size;
+ }
+ }
+ else {
+ mtype->buf = 0;
+ }
+ return mtype->buf;
+}
+
+/*
+ * Check value of received vector datatype buffer
+ */
+static int MTestTypeVectorCheckbuf(MTestDatatype * mtype)
+{
+ unsigned char *p;
+ unsigned char expected;
+ int i, err = 0, merr;
+ MPI_Aint size = 0, byte_offset, dt_offset, extent, lb;
+
+ p = (unsigned char *) mtype->buf;
+ if (p) {
+ MPI_Aint k, j;
+ int nc;
+ merr = MPI_Type_get_extent(mtype->datatype, &lb, &extent);
+ if (merr)
+ MTestPrintError(merr);
+
+ size = extent + lb;
+ nc = 0;
+ dt_offset = 0;
+ /* For each datatype */
+ for (k = 0; k < mtype->count; k++) {
+ /* For each block */
+ for (i = 0; i < mtype->nblock; i++) {
+ byte_offset = dt_offset + i * mtype->stride;
+ /* For each byte */
+ for (j = 0; j < mtype->blksize; j++) {
+ expected = (unsigned char) (0xff ^ (nc & 0xff));
+ if (p[byte_offset + j] != expected) {
+ err++;
+ if (mtype->printErrors && err < 10) {
+ printf("Data expected = %x but got p[%d,%ld] = %x\n", expected, i, j,
+ p[byte_offset + j]);
+ fflush(stdout);
+ }
+ }
+ nc++;
+ }
+ }
+ dt_offset += size;
+ }
+ }
+ return err;
+}
+
+
+/* ------------------------------------------------------------------------ */
+/* Datatype routines for indexed datatypes */
+/* ------------------------------------------------------------------------ */
+
+/*
+ * Initialize buffer of indexed datatype
+ */
+static void *MTestTypeIndexedInit(MTestDatatype * mtype)
+{
+ MPI_Aint extent = 0, lb = 0, size, totsize, dt_offset, offset;
+ int merr;
+
+ if (mtype->count > 0) {
+ unsigned char *p;
+ MPI_Aint k, b;
+ int i, j, nc;
+
+ /* Allocate buffer */
+ merr = MPI_Type_get_extent(mtype->datatype, &lb, &extent);
+ if (merr)
+ MTestPrintError(merr);
+
+ size = extent + lb;
+ totsize = size * mtype->count;
+
+ if (!mtype->buf) {
+ mtype->buf = (void *) malloc(totsize);
+ }
+ p = (unsigned char *) (mtype->buf);
+ if (!p) {
+ char errmsg[128] = { 0 };
+ sprintf(errmsg, "Out of memory in %s", __FUNCTION__);
+ MTestError(errmsg);
+ }
+
+ /* First, set to -1 */
+ for (k = 0; k < totsize; k++)
+ p[k] = 0xff;
+
+ /* Now, set the actual elements to the successive values.
+ * We require that the base type is a contiguous type */
+ nc = 0;
+ dt_offset = 0;
+ /* For each datatype */
+ for (k = 0; k < mtype->count; k++) {
+ /* For each block */
+ for (i = 0; i < mtype->nblock; i++) {
+ /* For each element in the block */
+ for (j = 0; j < mtype->index[i]; j++) {
+ offset = dt_offset + mtype->displ_in_bytes[i]
+ + j * mtype->basesize;
+ /* For each byte in the element */
+ for (b = 0; b < mtype->basesize; b++) {
+ p[offset + b] = (unsigned char) (0xff ^ (nc++ & 0xff));
+ }
+ }
+ }
+ dt_offset += size;
+ }
+ }
+ else {
+ /* count == 0 */
+ if (mtype->buf) {
+ free(mtype->buf);
+ }
+ mtype->buf = 0;
+ }
+ return mtype->buf;
+}
+
+/*
+ * Check value of received indexed datatype buffer
+ */
+static int MTestTypeIndexedCheckbuf(MTestDatatype * mtype)
+{
+ unsigned char *p;
+ unsigned char expected;
+ int err = 0, merr;
+ MPI_Aint size = 0, offset, dt_offset, extent = 0, lb = 0;
+
+ p = (unsigned char *) mtype->buf;
+ if (p) {
+ MPI_Aint k, b;
+ int i, j, nc;
+ merr = MPI_Type_get_extent(mtype->datatype, &lb, &extent);
+ if (merr)
+ MTestPrintError(merr);
+
+ size = lb + extent;
+ nc = 0;
+ dt_offset = 0;
+ /* For each datatype */
+ for (k = 0; k < mtype->count; k++) {
+ /* For each block */
+ for (i = 0; i < mtype->nblock; i++) {
+ /* For each element in the block */
+ for (j = 0; j < mtype->index[i]; j++) {
+ offset = dt_offset + mtype->displ_in_bytes[i]
+ + j * mtype->basesize;
+ /* For each byte in the element */
+ for (b = 0; b < mtype->basesize; b++) {
+ expected = (unsigned char) (0xff ^ (nc++ & 0xff));
+ if (p[offset + b] != expected) {
+ err++;
+ if (mtype->printErrors && err < 10) {
+ printf("Data expected = %x but got p[%d,%d] = %x\n",
+ expected, i, j, p[offset + b]);
+ fflush(stdout);
+ }
+ }
+ }
+ }
+ }
+ dt_offset += size;
+ }
+ }
+ return err;
+}
+
+/* ------------------------------------------------------------------------ */
+/* Datatype routines for indexed-block datatypes */
+/* ------------------------------------------------------------------------ */
+
+/*
+ * Initialize buffer of indexed-block datatype
+ */
+static void *MTestTypeIndexedBlockInit(MTestDatatype * mtype)
+{
+ MPI_Aint extent = 0, lb = 0, size, totsize, offset, dt_offset;
+ int merr;
+
+ if (mtype->count > 0) {
+ unsigned char *p;
+ MPI_Aint k, j;
+ int i, nc;
+
+ /* Allocate the send/recv buffer */
+ merr = MPI_Type_get_extent(mtype->datatype, &lb, &extent);
+ if (merr)
+ MTestPrintError(merr);
+ size = extent + lb;
+ totsize = size * mtype->count;
+
+ if (!mtype->buf) {
+ mtype->buf = (void *) malloc(totsize);
+ }
+ p = (unsigned char *) (mtype->buf);
+ if (!p) {
+ char errmsg[128] = { 0 };
+ sprintf(errmsg, "Out of memory in %s", __FUNCTION__);
+ MTestError(errmsg);
+ }
+
+ /* First, set to -1 */
+ for (k = 0; k < totsize; k++)
+ p[k] = 0xff;
+
+ /* Now, set the actual elements to the successive values.
+ * We require that the base type is a contiguous type */
+ nc = 0;
+ dt_offset = 0;
+ /* For each datatype */
+ for (k = 0; k < mtype->count; k++) {
+ /* For each block */
+ for (i = 0; i < mtype->nblock; i++) {
+ offset = dt_offset + mtype->displ_in_bytes[i];
+ /* For each byte in the block */
+ for (j = 0; j < mtype->blksize; j++) {
+ p[offset + j] = (unsigned char) (0xff ^ (nc++ & 0xff));
+ }
+ }
+ dt_offset += size;
+ }
+ }
+ else {
+ /* count == 0 */
+ if (mtype->buf) {
+ free(mtype->buf);
+ }
+ mtype->buf = 0;
+ }
+ return mtype->buf;
+}
+
+/*
+ * Check value of received indexed-block datatype buffer
+ */
+static int MTestTypeIndexedBlockCheckbuf(MTestDatatype * mtype)
+{
+ unsigned char *p;
+ unsigned char expected;
+ int err = 0, merr;
+ MPI_Aint size = 0, offset, dt_offset, lb = 0, extent = 0;
+
+ p = (unsigned char *) mtype->buf;
+ if (p) {
+ MPI_Aint j, k;
+ int i, nc;
+ merr = MPI_Type_get_extent(mtype->datatype, &lb, &extent);
+ if (merr)
+ MTestPrintError(merr);
+
+ size = lb + extent;
+ nc = 0;
+ dt_offset = 0;
+ /* For each datatype */
+ for (k = 0; k < mtype->count; k++) {
+ /* For each block */
+ for (i = 0; i < mtype->nblock; i++) {
+ offset = dt_offset + mtype->displ_in_bytes[i];
+ /* For each byte in the block */
+ for (j = 0; j < mtype->blksize; j++) {
+ expected = (unsigned char) (0xff ^ (nc++ & 0xff));
+ if (p[offset + j] != expected) {
+ err++;
+ if (mtype->printErrors && err < 10) {
+ printf("Data expected = %x but got p[%d,%ld] = %x\n",
+ expected, i, j, p[offset + j]);
+ fflush(stdout);
+ }
+ }
+ }
+ }
+ dt_offset += size;
+ }
+ }
+ return err;
+}
+
+/* ------------------------------------------------------------------------ */
+/* Datatype routines for subarray datatypes with order Fortran */
+/* ------------------------------------------------------------------------ */
+
+/*
+ * Initialize buffer of subarray datatype.
+ */
+static void *MTestTypeSubarrayInit(MTestDatatype * mtype)
+{
+ MPI_Aint extent = 0, lb = 0, size, totsize, offset, dt_offset, byte_offset;
+ int merr;
+
+ if (mtype->count > 0) {
+ unsigned char *p;
+ MPI_Aint k;
+ int j, b, i, nc;
+
+ /* Allocate the send/recv buffer */
+ merr = MPI_Type_get_extent(mtype->datatype, &lb, &extent);
+ if (merr)
+ MTestPrintError(merr);
+
+ size = extent + lb;
+ totsize = size * mtype->count;
+
+ if (!mtype->buf) {
+ mtype->buf = (void *) malloc(totsize);
+ }
+ p = (unsigned char *) (mtype->buf);
+ if (!p) {
+ char errmsg[128] = { 0 };
+ sprintf(errmsg, "Out of memory in %s", __FUNCTION__);
+ MTestError(errmsg);
+ }
+
+ /* First, set to -1 */
+ for (k = 0; k < totsize; k++)
+ p[k] = 0xff;
+
+ /* Now, set the actual elements to the successive values.
+ * We require that the base type is a contiguous type. */
+ int ncol, sub_ncol, sub_nrow, sub_col_start, sub_row_start;
+ ncol = mtype->arr_sizes[1];
+ sub_nrow = mtype->arr_subsizes[0];
+ sub_ncol = mtype->arr_subsizes[1];
+ sub_row_start = mtype->arr_starts[0];
+ sub_col_start = mtype->arr_starts[1];
+
+ nc = 0;
+ dt_offset = 0;
+ /* For each datatype */
+ for (k = 0; k < mtype->count; k++) {
+ /* For each row */
+ for (i = 0; i < sub_nrow; i++) {
+ offset = (sub_row_start + i) * ncol + sub_col_start;
+ /* For each element in row */
+ for (j = 0; j < sub_ncol; j++) {
+ byte_offset = dt_offset + (offset + j) * mtype->basesize;
+ /* For each byte in element */
+ for (b = 0; b < mtype->basesize; b++)
+ p[byte_offset + b] = (unsigned char) (0xff ^ (nc++ & 0xff));
+ }
+ }
+ dt_offset += size;
+ }
+ }
+ else {
+ /* count == 0 */
+ if (mtype->buf) {
+ free(mtype->buf);
+ }
+ mtype->buf = 0;
+ }
+ return mtype->buf;
+}
+
+/*
+ * Check value of received subarray datatype buffer
+ */
+static int MTestTypeSubarrayCheckbuf(MTestDatatype * mtype)
+{
+ unsigned char *p;
+ unsigned char expected;
+ int err = 0, merr;
+ MPI_Aint size, offset, dt_offset, byte_offset, lb = 0, extent = 0;
+
+ p = (unsigned char *) mtype->buf;
+ if (p) {
+ MPI_Aint k;
+ int j, b, i, nc;
+ merr = MPI_Type_get_extent(mtype->datatype, &lb, &extent);
+ if (merr)
+ MTestPrintError(merr);
+
+ size = lb + extent;
+
+ int ncol, sub_ncol, sub_nrow, sub_col_start, sub_row_start;
+ ncol = mtype->arr_sizes[1];
+ sub_nrow = mtype->arr_subsizes[0];
+ sub_ncol = mtype->arr_subsizes[1];
+ sub_row_start = mtype->arr_starts[0];
+ sub_col_start = mtype->arr_starts[1];
+
+ nc = 0;
+ dt_offset = 0;
+ /* For each datatype */
+ for (k = 0; k < mtype->count; k++) {
+ /* For each row */
+ for (i = 0; i < sub_nrow; i++) {
+ offset = (sub_row_start + i) * ncol + sub_col_start;
+ /* For each element in row */
+ for (j = 0; j < sub_ncol; j++) {
+ byte_offset = dt_offset + (offset + j) * mtype->basesize;
+ /* For each byte in element */
+ for (b = 0; b < mtype->basesize; b++) {
+ expected = (unsigned char) (0xff ^ (nc++ & 0xff));
+ if (p[byte_offset + b] != expected) {
+ err++;
+ if (mtype->printErrors && err < 10) {
+ printf("Data expected = %x but got p[%d,%d,%d] = %x\n",
+ expected, i, j, b, p[byte_offset + b]);
+ fflush(stdout);
+ }
+ }
+ }
+ }
+ }
+ dt_offset += size;
+ }
+ }
+ if (err)
+ printf("%s error\n", __FUNCTION__);
+ return err;
+}
+
+/* ------------------------------------------------------------------------ */
+/* Datatype creators */
+/* ------------------------------------------------------------------------ */
+
+/*
+ * Setup contiguous type info and handlers.
+ *
+ * A contiguous datatype is created by using following parameters (stride is unused).
+ * nblock: Number of blocks.
+ * blocklen: Number of elements in each block. The total number of elements in
+ * this datatype is set as (nblock * blocklen).
+ * lb: Lower bound of the new datatype (ignored).
+ * oldtype: Datatype of element.
+ */
+static int MTestTypeContiguousCreate(MPI_Aint nblock, MPI_Aint blocklen, MPI_Aint stride,
+ MPI_Aint lb, MPI_Datatype oldtype, const char *typename_prefix,
+ MTestDatatype * mtype)
+{
+ int merr = 0;
+ char type_name[128];
+
+ MTestTypeReset(mtype);
+
+ merr = MPI_Type_size(oldtype, &mtype->basesize);
+ if (merr)
+ MTestPrintError(merr);
+
+ mtype->nblock = nblock;
+ mtype->blksize = blocklen * mtype->basesize;
+
+ merr = MPI_Type_contiguous(nblock * blocklen, oldtype, &mtype->datatype);
+ if (merr)
+ MTestPrintError(merr);
+ merr = MPI_Type_commit(&mtype->datatype);
+ if (merr)
+ MTestPrintError(merr);
+
+ memset(type_name, 0, sizeof(type_name));
+ sprintf(type_name, "%s %s (%ld count)", typename_prefix, "contiguous", nblock * blocklen);
+ merr = MPI_Type_set_name(mtype->datatype, (char *) type_name);
+ if (merr)
+ MTestPrintError(merr);
+
+ mtype->InitBuf = MTestTypeContigInit;
+ mtype->FreeBuf = MTestTypeFree;
+ mtype->CheckBuf = MTestTypeContigCheckbuf;
+ return merr;
+}
+
+/*
+ * Setup vector type info and handlers.
+ *
+ * A vector datatype is created by using following parameters.
+ * nblock: Number of blocks.
+ * blocklen: Number of elements in each block.
+ * stride: Strided number of elements between blocks.
+ * lb: Lower bound of the new datatype (ignored).
+ * oldtype: Datatype of element.
+ */
+static int MTestTypeVectorCreate(MPI_Aint nblock, MPI_Aint blocklen, MPI_Aint stride, MPI_Aint lb,
+ MPI_Datatype oldtype, const char *typename_prefix,
+ MTestDatatype * mtype)
+{
+ int merr = 0;
+ char type_name[128];
+
+ MTestTypeReset(mtype);
+
+ merr = MPI_Type_size(oldtype, &mtype->basesize);
+ if (merr)
+ MTestPrintError(merr);
+
+ /* These sizes are in bytes (see the VectorInit code) */
+ mtype->stride = stride * mtype->basesize;
+ mtype->blksize = blocklen * mtype->basesize;
+ mtype->nblock = nblock;
+
+ /* Vector uses stride in oldtypes */
+ merr = MPI_Type_vector(nblock, blocklen, stride, oldtype, &mtype->datatype);
+ if (merr)
+ MTestPrintError(merr);
+ merr = MPI_Type_commit(&mtype->datatype);
+ if (merr)
+ MTestPrintError(merr);
+
+ memset(type_name, 0, sizeof(type_name));
+ sprintf(type_name, "%s %s (%ld nblock %ld blocklen %ld stride)", typename_prefix, "vector",
+ nblock, blocklen, stride);
+ merr = MPI_Type_set_name(mtype->datatype, (char *) type_name);
+ if (merr)
+ MTestPrintError(merr);
+
+ mtype->InitBuf = MTestTypeVectorInit;
+ mtype->FreeBuf = MTestTypeFree;
+ mtype->CheckBuf = MTestTypeVectorCheckbuf;
+ return merr;
+}
+
+/*
+ * Setup hvector type info and handlers.
+ *
+ * A hvector datatype is created by using following parameters.
+ * nblock: Number of blocks.
+ * blocklen: Number of elements in each block.
+ * stride: Strided number of elements between blocks.
+ * lb: Lower bound of the new datatype (ignored).
+ * oldtype: Datatype of element.
+ */
+static int MTestTypeHvectorCreate(MPI_Aint nblock, MPI_Aint blocklen, MPI_Aint stride, MPI_Aint lb,
+ MPI_Datatype oldtype, const char *typename_prefix,
+ MTestDatatype * mtype)
+{
+ int merr;
+ char type_name[128];
+
+ MTestTypeReset(mtype);
+
+ merr = MPI_Type_size(oldtype, &mtype->basesize);
+ if (merr)
+ MTestPrintError(merr);
+
+ /* These sizes are in bytes (see the VectorInit code) */
+ mtype->stride = stride * mtype->basesize;
+ mtype->blksize = blocklen * mtype->basesize;
+ mtype->nblock = nblock;
+
+ /* Hvector uses stride in bytes */
+ merr = MPI_Type_create_hvector(nblock, blocklen, mtype->stride, oldtype, &mtype->datatype);
+ if (merr)
+ MTestPrintError(merr);
+ merr = MPI_Type_commit(&mtype->datatype);
+ if (merr)
+ MTestPrintError(merr);
+
+ memset(type_name, 0, sizeof(type_name));
+ sprintf(type_name, "%s %s (%ld nblock %ld blocklen %ld stride)", typename_prefix, "hvector",
+ nblock, blocklen, stride);
+ merr = MPI_Type_set_name(mtype->datatype, (char *) type_name);
+ if (merr)
+ MTestPrintError(merr);
+
+ /* User the same functions as vector, because mtype->stride is in bytes */
+ mtype->InitBuf = MTestTypeVectorInit;
+ mtype->FreeBuf = MTestTypeFree;
+ mtype->CheckBuf = MTestTypeVectorCheckbuf;
+
+ return merr;
+}
+
+/*
+ * Setup indexed type info and handlers.
+ *
+ * A indexed datatype is created by using following parameters.
+ * nblock: Number of blocks.
+ * blocklen: Number of elements in each block. Each block has the same length.
+ * stride: Strided number of elements between two adjacent blocks. The
+ * displacement of each block is set as (index of current block * stride).
+ * lb: Lower bound of the new datatype.
+ * oldtype: Datatype of element.
+ */
+static int MTestTypeIndexedCreate(MPI_Aint nblock, MPI_Aint blocklen, MPI_Aint stride, MPI_Aint lb,
+ MPI_Datatype oldtype, const char *typename_prefix,
+ MTestDatatype * mtype)
+{
+ int merr = 0;
+ char type_name[128];
+ int i;
+
+ MTestTypeReset(mtype);
+
+ merr = MPI_Type_size(oldtype, &mtype->basesize);
+ if (merr)
+ MTestPrintError(merr);
+
+ mtype->displs = (int *) malloc(nblock * sizeof(int));
+ mtype->displ_in_bytes = (MPI_Aint *) malloc(nblock * sizeof(MPI_Aint));
+ mtype->index = (int *) malloc(nblock * sizeof(int));
+ if (!mtype->displs || !mtype->displ_in_bytes || !mtype->index) {
+ char errmsg[128] = { 0 };
+ sprintf(errmsg, "Out of memory in %s", __FUNCTION__);
+ MTestError(errmsg);
+ }
+
+ mtype->nblock = nblock;
+ for (i = 0; i < nblock; i++) {
+ mtype->index[i] = blocklen;
+ mtype->displs[i] = lb + stride * i; /*stride between the start of two blocks */
+ mtype->displ_in_bytes[i] = (lb + stride * i) * mtype->basesize;
+ }
+
+ /* Indexed uses displacement in oldtypes */
+ merr = MPI_Type_indexed(nblock, mtype->index, mtype->displs, oldtype, &mtype->datatype);
+ if (merr)
+ MTestPrintError(merr);
+ merr = MPI_Type_commit(&mtype->datatype);
+ if (merr)
+ MTestPrintError(merr);
+
+ memset(type_name, 0, sizeof(type_name));
+ sprintf(type_name, "%s %s (%ld nblock %ld blocklen %ld stride %ld lb)", typename_prefix,
+ "index", nblock, blocklen, stride, lb);
+ merr = MPI_Type_set_name(mtype->datatype, (char *) type_name);
+ if (merr)
+ MTestPrintError(merr);
+
+ mtype->InitBuf = MTestTypeIndexedInit;
+ mtype->FreeBuf = MTestTypeFree;
+ mtype->CheckBuf = MTestTypeIndexedCheckbuf;
+
+ return merr;
+}
+
+/*
+ * Setup hindexed type info and handlers.
+ *
+ * A hindexed datatype is created by using following parameters.
+ * nblock: Number of blocks.
+ * blocklen: Number of elements in each block. Each block has the same length.
+ * stride: Strided number of elements between two adjacent blocks. The byte
+ * displacement of each block is set as (index of current block * stride * size of oldtype).
+ * lb: Lower bound of the new datatype.
+ * oldtype: Datatype of element.
+ */
+static inline int MTestTypeHindexedCreate(MPI_Aint nblock, MPI_Aint blocklen, MPI_Aint stride,
+ MPI_Aint lb, MPI_Datatype oldtype,
+ const char *typename_prefix, MTestDatatype * mtype)
+{
+ int merr;
+ char type_name[128];
+ int i;
+
+ MTestTypeReset(mtype);
+
+ merr = MPI_Type_size(oldtype, &mtype->basesize);
+ if (merr)
+ MTestPrintError(merr);
+
+ mtype->index = (int *) malloc(nblock * sizeof(int));
+ mtype->displ_in_bytes = (MPI_Aint *) malloc(nblock * sizeof(MPI_Aint));
+ if (!mtype->displ_in_bytes || !mtype->index) {
+ char errmsg[128] = { 0 };
+ sprintf(errmsg, "Out of memory in %s", __FUNCTION__);
+ MTestError(errmsg);
+ }
+
+ mtype->nblock = nblock;
+ for (i = 0; i < nblock; i++) {
+ mtype->index[i] = blocklen;
+ mtype->displ_in_bytes[i] = (lb + stride * i) * mtype->basesize;
+ }
+
+ /* Hindexed uses displacement in bytes */
+ merr = MPI_Type_create_hindexed(nblock, mtype->index, mtype->displ_in_bytes,
+ oldtype, &mtype->datatype);
+ if (merr)
+ MTestPrintError(merr);
+ merr = MPI_Type_commit(&mtype->datatype);
+ if (merr)
+ MTestPrintError(merr);
+
+ memset(type_name, 0, sizeof(type_name));
+ sprintf(type_name, "%s %s (%ld nblock %ld blocklen %ld stride %ld lb)", typename_prefix,
+ "hindex", nblock, blocklen, stride, lb);
+ merr = MPI_Type_set_name(mtype->datatype, (char *) type_name);
+ if (merr)
+ MTestPrintError(merr);
+
+ /* Reuse indexed functions, because all of them only use displ_in_bytes */
+ mtype->InitBuf = MTestTypeIndexedInit;
+ mtype->FreeBuf = MTestTypeFree;
+ mtype->CheckBuf = MTestTypeIndexedCheckbuf;
+
+ return merr;
+}
+
+
+/*
+ * Setup indexed-block type info and handlers.
+ *
+ * A indexed-block datatype is created by using following parameters.
+ * nblock: Number of blocks.
+ * blocklen: Number of elements in each block.
+ * stride: Strided number of elements between two adjacent blocks. The
+ * displacement of each block is set as (index of current block * stride).
+ * lb: Lower bound of the new datatype.
+ * oldtype: Datatype of element.
+ */
+static int MTestTypeIndexedBlockCreate(MPI_Aint nblock, MPI_Aint blocklen, MPI_Aint stride,
+ MPI_Aint lb, MPI_Datatype oldtype,
+ const char *typename_prefix, MTestDatatype * mtype)
+{
+ int merr;
+ char type_name[128];
+ int i;
+
+ MTestTypeReset(mtype);
+
+ merr = MPI_Type_size(oldtype, &mtype->basesize);
+ if (merr)
+ MTestPrintError(merr);
+
+ mtype->displs = (int *) malloc(nblock * sizeof(int));
+ mtype->displ_in_bytes = (MPI_Aint *) malloc(nblock * sizeof(MPI_Aint));
+ if (!mtype->displs || !mtype->displ_in_bytes) {
+ char errmsg[128] = { 0 };
+ sprintf(errmsg, "Out of memory in %s", __FUNCTION__);
+ MTestError(errmsg);
+ }
+
+ mtype->nblock = nblock;
+ mtype->blksize = blocklen * mtype->basesize;
+ for (i = 0; i < nblock; i++) {
+ mtype->displs[i] = lb + stride * i;
+ mtype->displ_in_bytes[i] = (lb + stride * i) * mtype->basesize;
+ }
+
+ /* Indexed-block uses displacement in oldtypes */
+ merr = MPI_Type_create_indexed_block(nblock, blocklen, mtype->displs,
+ oldtype, &mtype->datatype);
+ if (merr)
+ MTestPrintError(merr);
+ merr = MPI_Type_commit(&mtype->datatype);
+ if (merr)
+ MTestPrintError(merr);
+
+ memset(type_name, 0, sizeof(type_name));
+ sprintf(type_name, "%s %s (%ld nblock %ld blocklen %ld stride %ld lb)", typename_prefix,
+ "index_block", nblock, blocklen, stride, lb);
+ merr = MPI_Type_set_name(mtype->datatype, (char *) type_name);
+ if (merr)
+ MTestPrintError(merr);
+
+ mtype->InitBuf = MTestTypeIndexedBlockInit;
+ mtype->FreeBuf = MTestTypeFree;
+ mtype->CheckBuf = MTestTypeIndexedBlockCheckbuf;
+
+ return merr;
+}
+
+/*
+ * Setup hindexed-block type info and handlers.
+ *
+ * A hindexed-block datatype is created by using following parameters.
+ * nblock: Number of blocks.
+ * blocklen: Number of elements in each block.
+ * stride: Strided number of elements between two adjacent blocks. The byte
+ * displacement of each block is set as (index of current block * stride * size of oldtype).
+ * lb: Lower bound of the new datatype.
+ * oldtype: Datatype of element.
+ */
+static int MTestTypeHindexedBlockCreate(MPI_Aint nblock, MPI_Aint blocklen, MPI_Aint stride,
+ MPI_Aint lb, MPI_Datatype oldtype,
+ const char *typename_prefix, MTestDatatype * mtype)
+{
+ int merr;
+ char type_name[128];
+ int i;
+
+ MTestTypeReset(mtype);
+
+ merr = MPI_Type_size(oldtype, &mtype->basesize);
+ if (merr)
+ MTestPrintError(merr);
+
+ mtype->displ_in_bytes = (MPI_Aint *) malloc(nblock * sizeof(MPI_Aint));
+ if (!mtype->displ_in_bytes) {
+ char errmsg[128] = { 0 };
+ sprintf(errmsg, "Out of memory in %s", __FUNCTION__);
+ MTestError(errmsg);
+ }
+
+ mtype->nblock = nblock;
+ mtype->blksize = blocklen * mtype->basesize;
+ for (i = 0; i < nblock; i++) {
+ mtype->displ_in_bytes[i] = (lb + stride * i) * mtype->basesize;
+ }
+
+ /* Hindexed-block uses displacement in bytes */
+ merr = MPI_Type_create_hindexed_block(nblock, blocklen, mtype->displ_in_bytes,
+ oldtype, &mtype->datatype);
+ if (merr)
+ MTestPrintError(merr);
+ merr = MPI_Type_commit(&mtype->datatype);
+ if (merr)
+ MTestPrintError(merr);
+
+ memset(type_name, 0, sizeof(type_name));
+ sprintf(type_name, "%s %s (%ld nblock %ld blocklen %ld stride %ld lb)", typename_prefix,
+ "hindex_block", nblock, blocklen, stride, lb);
+ merr = MPI_Type_set_name(mtype->datatype, (char *) type_name);
+ if (merr)
+ MTestPrintError(merr);
+
+ /* Reuse indexed-block functions, because all of them only use displ_in_bytes */
+ mtype->InitBuf = MTestTypeIndexedBlockInit;
+ mtype->FreeBuf = MTestTypeFree;
+ mtype->CheckBuf = MTestTypeIndexedBlockCheckbuf;
+
+ return merr;
+}
+
+/*
+ * Setup struct type info and handlers.
+ *
+ * A struct datatype is created by using following parameters.
+ * nblock: Number of blocks.
+ * blocklen: Number of elements in each block. Each block has the same length.
+ * stride: Strided number of elements between two adjacent blocks. The byte
+ * displacement of each block is set as (index of current block * stride * size of oldtype).
+ * lb: Lower bound of the new datatype.
+ * oldtype: Datatype of element. Each block has the same oldtype.
+ */
+static int MTestTypeStructCreate(MPI_Aint nblock, MPI_Aint blocklen, MPI_Aint stride, MPI_Aint lb,
+ MPI_Datatype oldtype, const char *typename_prefix,
+ MTestDatatype * mtype)
+{
+ int merr;
+ char type_name[128];
+ int i;
+
+ MTestTypeReset(mtype);
+
+ merr = MPI_Type_size(oldtype, &mtype->basesize);
+ if (merr)
+ MTestPrintError(merr);
+
+ mtype->old_datatypes = (MPI_Datatype *) malloc(nblock * sizeof(MPI_Datatype));
+ mtype->displ_in_bytes = (MPI_Aint *) malloc(nblock * sizeof(MPI_Aint));
+ mtype->index = (int *) malloc(nblock * sizeof(int));
+ if (!mtype->displ_in_bytes || !mtype->old_datatypes) {
+ char errmsg[128] = { 0 };
+ sprintf(errmsg, "Out of memory in %s", __FUNCTION__);
+ MTestError(errmsg);
+ }
+
+ mtype->nblock = nblock;
+ mtype->blksize = blocklen * mtype->basesize;
+ for (i = 0; i < nblock; i++) {
+ mtype->displ_in_bytes[i] = (lb + stride * i) * mtype->basesize;
+ mtype->old_datatypes[i] = oldtype;
+ mtype->index[i] = blocklen;
+ }
+
+ /* Struct uses displacement in bytes */
+ merr = MPI_Type_create_struct(nblock, mtype->index, mtype->displ_in_bytes,
+ mtype->old_datatypes, &mtype->datatype);
+ if (merr)
+ MTestPrintError(merr);
+ merr = MPI_Type_commit(&mtype->datatype);
+ if (merr)
+ MTestPrintError(merr);
+
+ memset(type_name, 0, sizeof(type_name));
+ sprintf(type_name, "%s %s (%ld nblock %ld blocklen %ld stride %ld lb)", typename_prefix,
+ "struct", nblock, blocklen, stride, lb);
+ merr = MPI_Type_set_name(mtype->datatype, (char *) type_name);
+ if (merr)
+ MTestPrintError(merr);
+
+ /* Reuse indexed functions, because they use the same displ_in_bytes and index */
+ mtype->InitBuf = MTestTypeIndexedInit;
+ mtype->FreeBuf = MTestTypeFree;
+ mtype->CheckBuf = MTestTypeIndexedCheckbuf;
+
+ return merr;
+}
+
+/*
+ * Setup order-C subarray type info and handlers.
+ *
+ * A 2D-subarray datatype specified with order C and located in the right-bottom
+ * of the full array is created by using input parameters.
+ * Number of elements in the dimensions of the full array: {nblock + lb, stride}
+ * Number of elements in the dimensions of the subarray: {nblock, blocklen}
+ * Starting of the subarray in each dimension: {1, stride - blocklen}
+ * order: MPI_ORDER_C
+ * oldtype: oldtype
+ */
+static int MTestTypeSubArrayOrderCCreate(MPI_Aint nblock, MPI_Aint blocklen, MPI_Aint stride,
+ MPI_Aint lb, MPI_Datatype oldtype,
+ const char *typename_prefix, MTestDatatype * mtype)
+{
+ int merr;
+ char type_name[128];
+
+ MTestTypeReset(mtype);
+
+ merr = MPI_Type_size(oldtype, &mtype->basesize);
+ if (merr)
+ MTestPrintError(merr);
+
+ mtype->arr_sizes[0] = nblock + lb; /* {row, col} */
+ mtype->arr_sizes[1] = stride;
+ mtype->arr_subsizes[0] = nblock; /* {row, col} */
+ mtype->arr_subsizes[1] = blocklen;
+ mtype->arr_starts[0] = lb; /* {row, col} */
+ mtype->arr_starts[1] = stride - blocklen;
+ mtype->order = MPI_ORDER_C;
+
+ merr = MPI_Type_create_subarray(2, mtype->arr_sizes, mtype->arr_subsizes, mtype->arr_starts,
+ mtype->order, oldtype, &mtype->datatype);
+ if (merr)
+ MTestPrintError(merr);
+ merr = MPI_Type_commit(&mtype->datatype);
+ if (merr)
+ MTestPrintError(merr);
+
+ memset(type_name, 0, sizeof(type_name));
+ sprintf(type_name, "%s %s (full{%d,%d}, sub{%d,%d},start{%d,%d})",
+ typename_prefix, "subarray-c", mtype->arr_sizes[0], mtype->arr_sizes[1],
+ mtype->arr_subsizes[0], mtype->arr_subsizes[1], mtype->arr_starts[0],
+ mtype->arr_starts[1]);
+ merr = MPI_Type_set_name(mtype->datatype, (char *) type_name);
+ if (merr)
+ MTestPrintError(merr);
+
+ mtype->InitBuf = MTestTypeSubarrayInit;
+ mtype->FreeBuf = MTestTypeFree;
+ mtype->CheckBuf = MTestTypeSubarrayCheckbuf;
+
+ return merr;
+}
+
+
+/*
+ * Setup order-Fortran subarray type info and handlers.
+ *
+ * A 2D-subarray datatype specified with order Fortran and located in the right
+ * bottom of the full array is created by using input parameters.
+ * Number of elements in the dimensions of the full array: {stride, nblock + lb}
+ * Number of elements in the dimensions of the subarray: {blocklen, nblock}
+ * Starting of the subarray in each dimension: {stride - blocklen, lb}
+ * order: MPI_ORDER_FORTRAN
+ * oldtype: oldtype
+ */
+static int MTestTypeSubArrayOrderFortranCreate(MPI_Aint nblock, MPI_Aint blocklen, MPI_Aint stride,
+ MPI_Aint lb, MPI_Datatype oldtype,
+ const char *typename_prefix, MTestDatatype * mtype)
+{
+ int merr;
+ char type_name[128];
+
+ MTestTypeReset(mtype);
+
+ merr = MPI_Type_size(oldtype, &mtype->basesize);
+ if (merr)
+ MTestPrintError(merr);
+
+ /* use the same row and col as that of order-c subarray for buffer
+ * initialization and check because we access buffer in order-c */
+ mtype->arr_sizes[0] = nblock + lb; /* {row, col} */
+ mtype->arr_sizes[1] = stride;
+ mtype->arr_subsizes[0] = nblock; /* {row, col} */
+ mtype->arr_subsizes[1] = blocklen;
+ mtype->arr_starts[0] = lb; /* {row, col} */
+ mtype->arr_starts[1] = stride - blocklen;
+ mtype->order = MPI_ORDER_FORTRAN;
+
+ /* reverse row and col when create datatype so that we can get the same
+ * packed data on the other side in order to reuse the contig check function */
+ int arr_sizes[2] = { mtype->arr_sizes[1], mtype->arr_sizes[0] };
+ int arr_subsizes[2] = { mtype->arr_subsizes[1], mtype->arr_subsizes[0] };
+ int arr_starts[2] = { mtype->arr_starts[1], mtype->arr_starts[0] };
+
+ merr = MPI_Type_create_subarray(2, arr_sizes, arr_subsizes, arr_starts,
+ mtype->order, oldtype, &mtype->datatype);
+ if (merr)
+ MTestPrintError(merr);
+ merr = MPI_Type_commit(&mtype->datatype);
+ if (merr)
+ MTestPrintError(merr);
+
+ memset(type_name, 0, sizeof(type_name));
+ sprintf(type_name, "%s %s (full{%d,%d}, sub{%d,%d},start{%d,%d})",
+ typename_prefix, "subarray-f", arr_sizes[0], arr_sizes[1],
+ arr_subsizes[0], arr_subsizes[1], arr_starts[0], arr_starts[1]);
+ merr = MPI_Type_set_name(mtype->datatype, (char *) type_name);
+ if (merr)
+ MTestPrintError(merr);
+
+ mtype->InitBuf = MTestTypeSubarrayInit;
+ mtype->FreeBuf = MTestTypeFree;
+ mtype->CheckBuf = MTestTypeSubarrayCheckbuf;
+
+ return merr;
+}
+
+/* ------------------------------------------------------------------------ */
+/* Datatype routines exposed to test generator */
+/* ------------------------------------------------------------------------ */
+
+/*
+ * Setup basic type info and handlers.
+ */
+int MTestTypeBasicCreate(MPI_Datatype oldtype, MTestDatatype * mtype)
+{
+ int merr = 0;
+
+ MTestTypeReset(mtype);
+
+ merr = MPI_Type_size(oldtype, &mtype->basesize);
+ if (merr)
+ MTestPrintError(merr);
+
+ mtype->datatype = oldtype;
+ mtype->isBasic = 1;
+ mtype->InitBuf = MTestTypeContigInit;
+ mtype->FreeBuf = MTestTypeFree;
+ mtype->CheckBuf = MTestTypeContigCheckbuf;
+
+ return merr;
+}
+
+/*
+ * Setup dup type info and handlers.
+ *
+ * A dup datatype is created by using following parameters.
+ * oldtype: Datatype of element.
+ */
+int MTestTypeDupCreate(MPI_Datatype oldtype, MTestDatatype * mtype)
+{
+ int merr = 0;
+
+ MTestTypeReset(mtype);
+
+ merr = MPI_Type_size(oldtype, &mtype->basesize);
+ if (merr)
+ MTestPrintError(merr);
+
+ merr = MPI_Type_dup(oldtype, &mtype->datatype);
+ if (merr)
+ MTestPrintError(merr);
+
+ /* dup'ed types are already committed if the original type
+ * was committed (MPI-2, section 8.8) */
+
+ mtype->InitBuf = MTestTypeContigInit;
+ mtype->FreeBuf = MTestTypeFree;
+ mtype->CheckBuf = MTestTypeContigCheckbuf;
+
+ return merr;
+}
+
+
+/*
+ * General initialization for receive buffer.
+ * Allocate buffer and initialize for reception (e.g., set initial data to detect failure)
+ * Both basic and derived datatype can be handled by using extent as buffer size.
+ */
+void *MTestTypeInitRecv(MTestDatatype * mtype)
+{
+ MPI_Aint size, extent = 0, lb = 0;
+ int merr;
+
+ if (mtype->count > 0) {
+ signed char *p;
+ MPI_Aint i, totsize;
+ merr = MPI_Type_get_extent(mtype->datatype, &lb, &extent);
+ if (merr)
+ MTestPrintError(merr);
+
+ size = extent + lb;
+ totsize = size * mtype->count;
+ if (!mtype->buf) {
+ mtype->buf = (void *) malloc(totsize);
+ }
+ p = (signed char *) (mtype->buf);
+ if (!p) {
+ char errmsg[128] = { 0 };
+ sprintf(errmsg, "Out of memory in %s", __FUNCTION__);
+ MTestError(errmsg);
+ }
+ for (i = 0; i < totsize; i++) {
+ p[i] = 0xff;
+ }
+ }
+ else {
+ if (mtype->buf) {
+ free(mtype->buf);
+ }
+ mtype->buf = 0;
+ }
+ return mtype->buf;
+}
+
+void MTestTypeCreatorInit(MTestDdtCreator * creators)
+{
+ memset(creators, 0, sizeof(MTestDdtCreator) * MTEST_DDT_MAX);
+ creators[MTEST_DDT_CONTIGUOUS] = MTestTypeContiguousCreate;
+ creators[MTEST_DDT_VECTOR] = MTestTypeVectorCreate;
+ creators[MTEST_DDT_HVECTOR] = MTestTypeHvectorCreate;
+ creators[MTEST_DDT_INDEXED] = MTestTypeIndexedCreate;
+ creators[MTEST_DDT_HINDEXED] = MTestTypeHindexedCreate;
+ creators[MTEST_DDT_INDEXED_BLOCK] = MTestTypeIndexedBlockCreate;
+ creators[MTEST_DDT_HINDEXED_BLOCK] = MTestTypeHindexedBlockCreate;
+ creators[MTEST_DDT_STRUCT] = MTestTypeStructCreate;
+/* creators[MTEST_DDT_SUBARRAY_ORDER_C] = MTestTypeSubArrayOrderCCreate;*/
+/* creators[MTEST_DDT_SUBARRAY_ORDER_FORTRAN] = MTestTypeSubArrayOrderFortranCreate;*/
+}
+
+void MTestTypeMinCreatorInit(MTestDdtCreator * creators)
+{
+ memset(creators, 0, sizeof(MTestDdtCreator) * MTEST_DDT_MAX);
+ creators[MTEST_MIN_DDT_VECTOR] = MTestTypeVectorCreate;
+ creators[MTEST_MIN_DDT_INDEXED] = MTestTypeIndexedCreate;
+}
