test/datatype: add test for short unpack on heteregeneous cluster

Signed-off-by: Gilles Gouaillardet <[email protected]>

Author: ggouaillardet

test/datatype/Makefile.am

@@ -18,7 +18,7 @@ if PROJECT_OMPI
     MPI_TESTS = checksum position position_noncontig ddt_test ddt_raw unpack_ooo ddt_pack external32
     MPI_CHECKS = to_self ddt_pack_hetero
 endif
-TESTS = opal_datatype_test $(MPI_TESTS)
+TESTS = opal_datatype_test unpack_hetero $(MPI_TESTS)
 
 check_PROGRAMS = $(TESTS) $(MPI_CHECKS)
 
@@ -48,7 +48,9 @@ ddt_pack_LDADD = \
 
 ddt_pack_hetero_SOURCES = ddt_pack_hetero.c
 ddt_pack_hetero_LDFLAGS = $(WRAPPER_EXTRA_LDFLAGS)
-ddt_pack_hetero_LDADD = $(top_builddir)/ompi/libmpi.la
+ddt_pack_hetero_LDADD = \
+        $(top_builddir)/ompi/lib@[email protected] \
+        $(top_builddir)/opal/lib@[email protected]
 
 checksum_SOURCES = checksum.c
 checksum_LDFLAGS = $(OMPI_PKG_CONFIG_LDFLAGS)
@@ -83,5 +85,10 @@ external32_LDADD = \
         $(top_builddir)/ompi/lib@[email protected] \
         $(top_builddir)/opal/lib@[email protected]
 
+unpack_hetero_SOURCES = unpack_hetero.c
+unpack_hetero_LDFLAGS = $(OMPI_PKG_CONFIG_LDFLAGS)
+unpack_hetero_LDADD = \
+        $(top_builddir)/opal/lib@[email protected]
+
 distclean:
 	rm -rf *.dSYM .deps .libs *.log *.o *.trs $(check_PROGRAMS) Makefile

test/datatype/unpack_hetero.c

@@ -0,0 +1,103 @@
+/* -*- Mode: C; c-basic-offset:4 ; -*- */
+/*
+ * Copyright (c) 2014-2017 Research Organization for Information Science
+ *                         and Technology (RIST). All rights reserved.
+ * $COPYRIGHT$
+ *
+ * Additional copyrights may follow
+ *
+ * $HEADER$
+ */
+
+#include "opal_config.h"
+#include "opal/runtime/opal.h"
+#include "opal/datatype/opal_datatype.h"
+#include "opal/datatype/opal_datatype_internal.h"
+#include "opal/datatype/opal_convertor.h"
+#include "opal/datatype/opal_datatype_prototypes.h"
+#include "opal/util/arch.h"
+#include <time.h>
+#include <stdlib.h>
+#ifdef HAVE_SYS_TIME_H
+#include <sys/time.h>
+#endif
+#include <stdio.h>
+#include <string.h>
+
+/* Compile with:
+gcc -DHAVE_CONFIG_H -I. -I../../include -I../.. -I../../include -I../../../ompi-trunk/opal -I../../../ompi-trunk/orte -g opal_datatype_test.c -o opal_datatype_test
+*/
+
+uint32_t remote_arch = 0xffffffff;
+
+/**
+ * Main function. Call several tests and print-out the results. It try to stress the convertor
+ * using difficult data-type constructions as well as strange segment sizes for the conversion.
+ * Usually, it is able to detect most of the data-type and convertor problems. Any modifications
+ * on the data-type engine should first pass all the tests from this file, before going into other
+ * tests.
+ */
+int main( int argc, char* argv[] )
+{
+#if OPAL_ENABLE_HETEROGENEOUS_SUPPORT
+    opal_datatype_init();
+
+    /**
+     * By default simulate homogeneous architectures.
+     */
+    remote_arch = opal_local_arch ^ OPAL_ARCH_ISBIGENDIAN;
+
+    opal_convertor_t * pConv;
+    int sbuf[2], rbuf[2];
+    size_t max_data;
+    struct iovec a;
+    uint32_t iov_count;
+
+    sbuf[0] = 0x01000000; sbuf[1] = 0x02000000;
+
+    printf( "\n\n#\n * TEST UNPACKING 1 int out of 1\n#\n\n" );
+
+    pConv = opal_convertor_create( remote_arch, 0 );
+    rbuf[0] = -1; rbuf[1] = -1;
+    if( OPAL_SUCCESS != opal_convertor_prepare_for_recv( pConv, &opal_datatype_int4, 1, rbuf ) ) {
+        printf( "Cannot attach the datatype to a convertor\n" );
+        return OPAL_ERROR;
+    }
+    
+    a.iov_base = sbuf;
+    a.iov_len = 4;
+    iov_count = 1;
+    max_data = 4;
+    opal_unpack_general( pConv, &a, &iov_count, &max_data );
+
+    assert(1 == rbuf[0]);
+    assert(-1 == rbuf[1]);
+    OBJ_RELEASE(pConv);
+
+    printf( "\n\n#\n * TEST UNPACKING 1 int out of 2\n#\n\n" );
+    pConv = opal_convertor_create( remote_arch, 0 );
+    rbuf[0] = -1; rbuf[1] = -1;
+    if( OPAL_SUCCESS != opal_convertor_prepare_for_recv( pConv, &opal_datatype_int4, 2, rbuf ) ) {
+        printf( "Cannot attach the datatype to a convertor\n" );
+        return OPAL_ERROR;
+    }
+    
+
+    a.iov_base = sbuf;
+    a.iov_len = 4;
+    iov_count = 1;
+    max_data = 4;
+    opal_unpack_general( pConv, &a, &iov_count, &max_data );
+
+    assert(1 == rbuf[0]);
+    assert(-1 == rbuf[1]);
+    OBJ_RELEASE(pConv);
+
+    /* clean-ups all data allocations */
+    opal_datatype_finalize();
+    opal_finalize();
+    return OPAL_SUCCESS;
+#else
+    return 77;
+#endif
+}