1ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 2ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown/* A test program to check whether the type-traversal functions in 3ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown mpiwrap.c (walk_type, walk_type_array) are correct. It does this 4ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown by sending a message to itself, thereby discovering what areas of 5ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown memory the MPI implementation itself believe constitute the type. 6ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown It then gets walk_type to enumerate the type, and compares the 7ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown results. */ 8ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 9ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown#include <stdio.h> 10ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown#include <stdlib.h> 11ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown#include <string.h> 12ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown#include <assert.h> 13ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown#include "mpi.h" 14ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown#include "../memcheck/memcheck.h" 15ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 16ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Browntypedef MPI_Datatype Ty; 17ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 18ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Browntypedef unsigned char Bool; 19ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown#define False ((Bool)0) 20ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown#define True ((Bool)1) 21ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 22ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brownvoid* walk_type_fn = NULL; 23ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 24ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brownstatic Ty tycon_Contiguous ( int count, Ty t ) 25ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown{ 26ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown Ty t2; 27ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown int r = MPI_Type_contiguous( count, t, &t2 ); 28ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown assert(r == MPI_SUCCESS); 29ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown return t2; 30ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown} 31ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 32ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brownstatic Ty tycon_Struct2 ( int d1, int copies1, Ty t1, 33ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown int d2, int copies2, Ty t2 ) 34ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown{ 35ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown int blocklens[2]; 36ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown MPI_Aint disps[2]; 37ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown Ty tys[2]; 38ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown Ty tres; 39ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown int r; 40ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown blocklens[0] = copies1; 41ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown blocklens[1] = copies2; 42ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown disps[0] = d1; 43ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown disps[1] = d2; 44ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown tys[0] = t1; 45ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown tys[1] = t2; 46ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown r = MPI_Type_struct( 2, blocklens, disps, tys, &tres ); 47ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown assert(r == MPI_SUCCESS); 48ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown return tres; 49ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown} 50ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 51ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brownstatic Ty tycon_Vector ( int count, int blocklen, int stride, Ty t ) 52ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown{ 53ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown Ty tres; 54ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown int r; 55ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown r = MPI_Type_vector( count, blocklen, stride, t, &tres ); 56ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown assert(r == MPI_SUCCESS); 57ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown return tres; 58ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown} 59ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 60ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brownstatic Ty tycon_HVector ( int count, int blocklen, MPI_Aint stride, Ty t ) 61ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown{ 62ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown Ty tres; 63ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown int r; 64ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown r = MPI_Type_hvector( count, blocklen, stride, t, &tres ); 65ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown assert(r == MPI_SUCCESS); 66ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown return tres; 67ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown} 68ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 69ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brownstatic Ty tycon_Indexed2 ( int d1, int copies1, 70ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown int d2, int copies2, Ty t ) 71ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown{ 72ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown int blocklens[2]; 73ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown int disps[2]; 74ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown Ty tres; 75ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown int r; 76ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown blocklens[0] = copies1; 77ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown blocklens[1] = copies2; 78ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown disps[0] = d1; 79ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown disps[1] = d2; 80ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown r = MPI_Type_indexed( 2, blocklens, disps, t, &tres ); 81ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown assert(r == MPI_SUCCESS); 82ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown return tres; 83ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown} 84ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 85ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brownstatic Ty tycon_HIndexed2 ( MPI_Aint d1, int copies1, 86ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown MPI_Aint d2, int copies2, Ty t ) 87ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown{ 88ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown int blocklens[2]; 89ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown MPI_Aint disps[2]; 90ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown Ty tres; 91ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown int r; 92ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown blocklens[0] = copies1; 93ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown blocklens[1] = copies2; 94ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown disps[0] = d1; 95ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown disps[1] = d2; 96ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown r = MPI_Type_hindexed( 2, blocklens, disps, t, &tres ); 97ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown assert(r == MPI_SUCCESS); 98ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown return tres; 99ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown} 100ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 101ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown/* ------------------------------ */ 102ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 103ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brownchar characterise ( unsigned char b ) 104ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown{ 105ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown if (b == 0x00) return 'D'; 106ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown if (b == 0xFF) return '.'; 107ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown return '?'; 108ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown} 109ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 110ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brownvoid sendToMyself_callback( void* v, long n ) 111ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown{ 112ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown long i; 113ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown unsigned char* p = (unsigned char*)v; 114ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown if (0) printf("callback: %p %ld\n", v, n); 115ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown for (i = 0; i < n; i++) 116ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown p[i] = 0x00; 117ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown} 118ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 119ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brownvoid sendToMyself ( Bool commit_free, Ty* tyP, char* name ) 120ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown{ 121ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown int i; 122ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown MPI_Aint lb, ub, ex; 123ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown MPI_Request req; 124ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown MPI_Status status; 125ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown char* sbuf; 126ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown char* rbuf; 127ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown char* rbuf_walk; 128ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown int r; 129ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 130ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown /* C: what a fabulous functional programming language :-) */ 131ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown void(*dl_walk_type)(void(*)(void*,long),char*,MPI_Datatype) 132ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown = (void(*)(void(*)(void*,long),char*,MPI_Datatype)) 133ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown walk_type_fn; 134ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 135ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown if (!dl_walk_type) { 136ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown printf("sendToMyself: can't establish type walker fn\n"); 137ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown return; 138ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown } 139ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 140ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown printf("\nsendToMyself: trying %s\n", name); 141ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 142ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown if (commit_free) { 143ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown r = MPI_Type_commit( tyP ); 144ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown assert(r == MPI_SUCCESS); 145ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown } 146ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 147ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown r = MPI_Type_lb( *tyP, &lb ); 148ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown assert(r == MPI_SUCCESS); 149ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown r = MPI_Type_ub( *tyP, &ub ); 150ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown assert(r == MPI_SUCCESS); 151ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown r = MPI_Type_extent( *tyP, &ex ); 152ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown assert(r == MPI_SUCCESS); 153ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown printf("sendToMyself: ex=%d (%d,%d)\n", (int)ex, (int)lb, (int)ub); 154ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown assert(lb >= 0); 155ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 156ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown /* Fill send buffer with zeroes */ 157ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown sbuf = malloc(ub); 158ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown assert(sbuf); 159ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown for (i = 0; i < ub; i++) 160ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown sbuf[i] = 0; 161ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 162ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown r = MPI_Isend( sbuf,1,*tyP, 0,99,MPI_COMM_WORLD, &req); 163ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown assert(r == MPI_SUCCESS); 164ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 165ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown /* Fill recv buffer with 0xFFs */ 166ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown rbuf = malloc(ub); 167ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown assert(rbuf); 168ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown for (i = 0; i < ub; i++) 169ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown rbuf[i] = 0xFF; 170ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 171ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown r = MPI_Recv( rbuf,1,*tyP, 0,99,MPI_COMM_WORLD, &status); 172ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown assert(r == MPI_SUCCESS); 173ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 174ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown /* Now: rbuf should contain 0x00s where data was transferred and 175ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown undefined 0xFFs where data was not transferred. Get 176ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown libmpiwrap.so to walk the transferred type, using the callback 177ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown to set to 0x00 all parts of rbuf_walk it considers part of the 178ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown type. */ 179ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 180ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown rbuf_walk = malloc(ub); 181ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown assert(rbuf_walk); 182ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown for (i = 0; i < ub; i++) 183ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown rbuf_walk[i] = 0xFF; 184ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 185ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown dl_walk_type( sendToMyself_callback, rbuf_walk, *tyP ); 186ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 187ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown if (commit_free) { 188ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown r = MPI_Type_free( tyP ); 189ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown assert(r == MPI_SUCCESS); 190ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown } 191ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 192ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown for (i = 0; i < ub; i++) { 193ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown if (rbuf_walk[i] == rbuf[i]) 194ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown continue; /* ok */ 195ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown else 196ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown break; /* discrepancy */ 197ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown } 198ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 199ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown if (i == ub) 200ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown printf("SUCCESS\n"); 201ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown else 202ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown printf("FAILED\n"); 203ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 204ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown printf(" libmpiwrap="); 205ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown for (i = 0; i < ub; i++) 206ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown printf("%c", characterise(rbuf_walk[i])); 207ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown printf("\n"); 208ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 209ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown printf("MPI library="); 210ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown for (i = 0; i < ub; i++) 211ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown printf("%c", characterise(rbuf[i])); 212ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown printf("\n"); 213ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 214ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown free(sbuf); 215ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown free(rbuf); 216ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown free(rbuf_walk); 217ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown} 218ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 219ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 220ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Browntypedef char* Nm; 221ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 222ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brownint main ( int argc, char** argv ) 223ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown{ 224ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown int rank, size; 225ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown char* opts; 226ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 227ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown if (!RUNNING_ON_VALGRIND) { 228ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown printf("error: this program must be run on valgrind\n"); 229ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown return 1; 230ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown } 231ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown opts = getenv("MPIWRAP_DEBUG"); 232ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown if ((!opts) || NULL==strstr(opts, "initkludge")) { 233ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown printf("error: program requires MPIWRAP_DEBUG=initkludge\n"); 234ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown return 1; 235ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown } 236ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 237ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown /* Note: this trick doesn't work on 64-bit platforms, 238ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown since MPI_Init returns int. */ 239ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown walk_type_fn = (void*)(long) MPI_Init( &argc, &argv ); 240ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown printf("mpiwrap_type_test: walk_type_fn = %p\n", walk_type_fn); 241ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown assert(walk_type_fn); 242ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 243ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown MPI_Comm_size( MPI_COMM_WORLD, &size ); 244ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown MPI_Comm_rank( MPI_COMM_WORLD, &rank ); 245ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 246ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown if (rank == 0) { 247ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 248ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown#define TRY(_commit_free,_type,_name) \ 249ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown do { Ty ty = (_type); \ 250ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown Nm nm = (_name); \ 251ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown sendToMyself((_commit_free), &ty, nm); \ 252ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown } while (0) 253ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 254ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown TRY(True, tycon_Contiguous(3, MPI_INT), 255ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown "Contig{3xINT}"); 256ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 257ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown TRY(True, tycon_Struct2(3,2,MPI_CHAR, 8,1,MPI_DOUBLE), 258ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown "Struct{h3:2xCHAR, h8:1xDOUBLE}"); 259ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 260ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown TRY(True, tycon_Struct2(0,1,MPI_CHAR, 8,1,tycon_Contiguous(4, MPI_DOUBLE)), 261ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown "Struct{h0:1xCHAR, h8:1xContig{4xDOUBLE}}"); 262ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 263ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown TRY(True, tycon_Contiguous(10, tycon_Struct2(1,1,MPI_CHAR, 4,1,MPI_FLOAT)), 264ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown "Contig{10xStruct{h1:1xCHAR, h4:1xFLOAT}}"); 265ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 266ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown TRY(True, tycon_Vector(5, 2,3,MPI_DOUBLE), 267ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown "Vector{5x(2,3)xDOUBLE}"); 268ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 269ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown TRY(True, tycon_Vector(3, 1,2,MPI_LONG_DOUBLE), 270ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown "Vector{3x(1,2)xLONG_DOUBLE}"); 271ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 272ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown TRY(True, tycon_HVector(4, 1,3,MPI_SHORT), 273ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown "HVector{4x(1,h3)xSHORT}"); 274ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 275ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown TRY(True, tycon_Indexed2(1,3, 5,2, MPI_UNSIGNED_CHAR), 276ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown "Indexed{1:3x,5:2x,UNSIGNED_CHAR}"); 277ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 278ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown TRY(True, tycon_HIndexed2(1,2, 6,3, MPI_UNSIGNED_SHORT), 279ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown "HIndexed{h1:2x,h6:3x,UNSIGNED_SHORT}"); 280ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 281ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown TRY(False, MPI_FLOAT_INT, "FLOAT_INT"); 282ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown TRY(False, MPI_DOUBLE_INT, "DOUBLE_INT"); 283ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown TRY(False, MPI_LONG_INT, "LONG_INT"); 284ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown TRY(False, MPI_SHORT_INT, "SHORT_INT"); 285ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown TRY(False, MPI_2INT, "2INT"); 286ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown TRY(False, MPI_LONG_DOUBLE_INT, "LONG_DOUBLE_INT"); 287ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 288ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown /* The next 4 don't seem to exist on openmpi-1.2.2. */ 289ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 290ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown#if defined(MPI_REAL8) 291ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown TRY(False, MPI_REAL8, "REAL8"); 292ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown#endif 293ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown#if defined(MPI_REAL4) 294ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown TRY(False, MPI_REAL4, "REAL4"); 295ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown#endif 296ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown#if defined(MPI_INTEGER8) 297ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown TRY(False, MPI_INTEGER8, "INTEGER8"); 298ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown#endif 299ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown#if defined(MPI_INTEGER4) 300ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown TRY(False, MPI_INTEGER4, "INTEGER4"); 301ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown#endif 302ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 303ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown TRY(False, MPI_COMPLEX, "COMPLEX"); 304ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown TRY(False, MPI_DOUBLE_COMPLEX, "DOUBLE_COMPLEX"); 305ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 306ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown // On openmpi-1.2.2 on x86-linux, sendToMyself bombs openmpi, 307ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown // for some reason (openmpi thinks these all have zero size/extent 308ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown // and therefore can't be MPI_Send-ed, AIUI). 309ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown // TRY(False, MPI_LOGICAL, "LOGICAL"); 310ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown // TRY(False, MPI_REAL, "REAL"); 311ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown // TRY(False, MPI_DOUBLE_PRECISION, "DOUBLE_PRECISION"); 312ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown // TRY(False, MPI_INTEGER, "INTEGER"); 313ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown TRY(False, MPI_2INTEGER, "2INTEGER"); 314ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown TRY(False, MPI_2COMPLEX, "2COMPLEX"); 315ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown TRY(False, MPI_2DOUBLE_COMPLEX, "2DOUBLE_COMPLEX"); 316ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown TRY(False, MPI_2REAL, "2REAL"); 317ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown TRY(False, MPI_2DOUBLE_PRECISION, "2DOUBLE_PRECISION"); 318ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown TRY(False, MPI_CHARACTER, "CHARACTER"); 319ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 320ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown /* The following from a table in chapter 9 of the MPI2 spec 321ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown date Nov 15, 2003, page 247. */ 322ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown TRY(False, MPI_PACKED, "PACKED"); 323ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown TRY(False, MPI_BYTE, "BYTE"); 324ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown TRY(False, MPI_CHAR, "CHAR"); 325ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown TRY(False, MPI_UNSIGNED_CHAR, "UNSIGNED_CHAR"); 326ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown TRY(False, MPI_SIGNED_CHAR, "SIGNED_CHAR"); 327ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown TRY(False, MPI_WCHAR, "WCHAR"); 328ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown TRY(False, MPI_SHORT, "SHORT"); 329ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown TRY(False, MPI_UNSIGNED_SHORT, "UNSIGNED_SHORT"); 330ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown TRY(False, MPI_INT, "INT"); 331ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown TRY(False, MPI_UNSIGNED, "UNSIGNED"); 332ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown TRY(False, MPI_LONG, "LONG"); 333ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown TRY(False, MPI_UNSIGNED_LONG, "UNSIGNED_LONG"); 334ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown TRY(False, MPI_FLOAT, "FLOAT"); 335ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown TRY(False, MPI_DOUBLE, "DOUBLE"); 336ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown TRY(False, MPI_LONG_DOUBLE, "LONG_DOUBLE"); 337ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown TRY(False, MPI_CHARACTER, "CHARACTER"); 338ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 339ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown // Same deal as above 340ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown // TRY(False, MPI_LOGICAL, "LOGICAL"); 341ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown // TRY(False, MPI_INTEGER, "INTEGER"); 342ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown // TRY(False, MPI_REAL, "REAL"); 343ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown // TRY(False, MPI_DOUBLE_PRECISION, "DOUBLE_PRECISION"); 344ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 345ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown TRY(False, MPI_COMPLEX, "COMPLEX"); 346ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown TRY(False, MPI_DOUBLE_COMPLEX, "DOUBLE_COMPLEX"); 347ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown#if defined(MPI_INTEGER1) 348ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown TRY(False, MPI_INTEGER1, "INTEGER1"); 349ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown#endif 350ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown#if defined(MPI_INTEGER2) 351ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown TRY(False, MPI_INTEGER2, "INTEGER2"); 352ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown#endif 353ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown#if defined(MPI_INTEGER4) 354ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown TRY(False, MPI_INTEGER4, "INTEGER4"); 355ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown#endif 356ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown#if defined(MPI_INTEGER8) 357ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown TRY(False, MPI_INTEGER8, "INTEGER8"); 358ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown#endif 359ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown TRY(False, MPI_LONG_LONG, "LONG_LONG"); 360ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown TRY(False, MPI_UNSIGNED_LONG_LONG, "UNSIGNED_LONG_LONG"); 361ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown#if defined(MPI_REAL4) 362ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown TRY(False, MPI_REAL4, "REAL4"); 363ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown#endif 364ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown#if defined(MPI_REAL8) 365ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown TRY(False, MPI_REAL8, "REAL8"); 366ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown#endif 367ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown#if defined(MPI_REAL16) 368ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown TRY(False, MPI_REAL16, "REAL16"); 369ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown#endif 370ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 371ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown#undef TRY 372ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 373ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown } 374ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown 375ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown MPI_Finalize(); 376ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown return 0; 377ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown} 378