int MPIX_Mrecv_x(void *buf, MPI_Count count, MPI_Datatype datatype, MPI_Message *message, MPI_Status *status)
{
int rc = MPI_SUCCESS;
if (likely (count <= bigmpi_int_max )) {
rc = MPI_Mrecv(buf, (int)count, datatype, message, status);
} else {
MPI_Datatype newtype;
BigMPI_Type_contiguous(0,count, datatype, &newtype);
MPI_Type_commit(&newtype);
rc = MPI_Mrecv(buf, 1, newtype, message, status);
MPI_Type_free(&newtype);
}
return rc;
}
FORT_DLL_SPEC void FORT_CALL mpi_mrecv_ ( void*v1, MPI_Fint *v2, MPI_Fint *v3, MPI_Fint *v4, MPI_Fint *v5, MPI_Fint *ierr ){
#ifndef HAVE_MPI_F_INIT_WORKS_WITH_C
if (MPIR_F_NeedInit){ mpirinitf_(); MPIR_F_NeedInit = 0; }
#endif
if (v5 == MPI_F_STATUS_IGNORE) { v5 = (MPI_Fint*)MPI_STATUS_IGNORE; }
*ierr = MPI_Mrecv( v1, (int)*v2, (MPI_Datatype)(*v3), (MPI_Message *)(v4), (MPI_Status *)v5 );
}
JNIEXPORT jlong JNICALL Java_mpi_Message_mRecv(
JNIEnv *env, jobject jthis, jlong jMessage, jobject buf, jboolean db,
jint off, jint count, jlong jType, jint bType, jlongArray jStatus)
{
MPI_Message message = (MPI_Message)jMessage;
MPI_Datatype type = (MPI_Datatype)jType;
void *ptr;
ompi_java_buffer_t *item;
ompi_java_getWritePtr(&ptr, &item, env, buf, db, count, type);
MPI_Status status;
int rc = MPI_Mrecv(ptr, count, type, &message, &status);
ompi_java_exceptionCheck(env, rc);
ompi_java_status_set(env, jStatus, &status);
ompi_java_releaseWritePtr(ptr, item, env, buf, db, off, count, type, bType);
return (jlong)message;
}
int MPIR_Mrecv_cdesc(CFI_cdesc_t* x0, int x1, MPI_Datatype x2, MPI_Message * x3, MPI_Status * x4)
{
int err = MPI_SUCCESS;
void *buf0 = x0->base_addr;
int count0 = x1;
MPI_Datatype dtype0 = x2;
if (buf0 == &MPIR_F08_MPI_BOTTOM) {
buf0 = MPI_BOTTOM;
}
if (x0->rank != 0 && !CFI_is_contiguous(x0)) {
err = cdesc_create_datatype(x0, x1, x2, &dtype0);
count0 = 1;
}
err = MPI_Mrecv(buf0, count0, dtype0, x3, x4);
if (dtype0 != x2) MPI_Type_free(&dtype0);
return err;
}
int main(int argc, char **argv)
{
int errs = 0;
int found, completed;
int rank, size;
int *sendbuf = NULL, *recvbuf = NULL;
int count, i;
MPI_Message msg;
MPI_Request rreq;
MPI_Status s1, s2;
MPI_Datatype vectype;
MTest_Init(&argc, &argv);
MPI_Comm_rank(MPI_COMM_WORLD, &rank);
MPI_Comm_size(MPI_COMM_WORLD, &size);
if (size < 2) {
printf("this test requires at least 2 processes\n");
MPI_Abort(MPI_COMM_WORLD, 1);
}
/* all processes besides ranks 0 & 1 aren't used by this test */
if (rank >= 2) {
goto epilogue;
}
sendbuf = (int *) malloc(LARGE_SZ * sizeof(int));
recvbuf = (int *) malloc(LARGE_SZ * sizeof(int));
if (sendbuf == NULL || recvbuf == NULL) {
printf("Error in memory allocation\n");
MPI_Abort(MPI_COMM_WORLD, 1);
}
/* test 0: simple send & mprobe+mrecv */
if (rank == 0) {
sendbuf[0] = 0xdeadbeef;
sendbuf[1] = 0xfeedface;
MPI_Send(sendbuf, 2, MPI_INT, 1, 5, MPI_COMM_WORLD);
} else {
memset(&s1, 0xab, sizeof(MPI_Status));
memset(&s2, 0xab, sizeof(MPI_Status));
/* the error field should remain unmodified */
s1.MPI_ERROR = MPI_ERR_DIMS;
s2.MPI_ERROR = MPI_ERR_TOPOLOGY;
msg = MPI_MESSAGE_NULL;
MPI_Mprobe(0, 5, MPI_COMM_WORLD, &msg, &s1);
check(s1.MPI_SOURCE == 0);
check(s1.MPI_TAG == 5);
check(s1.MPI_ERROR == MPI_ERR_DIMS);
check(msg != MPI_MESSAGE_NULL);
count = -1;
MPI_Get_count(&s1, MPI_INT, &count);
check(count == 2);
recvbuf[0] = 0x01234567;
recvbuf[1] = 0x89abcdef;
MPI_Mrecv(recvbuf, count, MPI_INT, &msg, &s2);
check(recvbuf[0] == 0xdeadbeef);
check(recvbuf[1] == 0xfeedface);
check(s2.MPI_SOURCE == 0);
check(s2.MPI_TAG == 5);
check(s2.MPI_ERROR == MPI_ERR_TOPOLOGY);
check(msg == MPI_MESSAGE_NULL);
}
/* test 1: simple send & mprobe+imrecv */
if (rank == 0) {
sendbuf[0] = 0xdeadbeef;
sendbuf[1] = 0xfeedface;
MPI_Send(sendbuf, 2, MPI_INT, 1, 5, MPI_COMM_WORLD);
} else {
memset(&s1, 0xab, sizeof(MPI_Status));
memset(&s2, 0xab, sizeof(MPI_Status));
/* the error field should remain unmodified */
s1.MPI_ERROR = MPI_ERR_DIMS;
s2.MPI_ERROR = MPI_ERR_TOPOLOGY;
msg = MPI_MESSAGE_NULL;
MPI_Mprobe(0, 5, MPI_COMM_WORLD, &msg, &s1);
check(s1.MPI_SOURCE == 0);
check(s1.MPI_TAG == 5);
check(s1.MPI_ERROR == MPI_ERR_DIMS);
check(msg != MPI_MESSAGE_NULL);
count = -1;
MPI_Get_count(&s1, MPI_INT, &count);
check(count == 2);
rreq = MPI_REQUEST_NULL;
recvbuf[0] = 0x01234567;
recvbuf[1] = 0x89abcdef;
MPI_Imrecv(recvbuf, count, MPI_INT, &msg, &rreq);
check(rreq != MPI_REQUEST_NULL);
MPI_Wait(&rreq, &s2);
check(recvbuf[0] == 0xdeadbeef);
check(recvbuf[1] == 0xfeedface);
check(s2.MPI_SOURCE == 0);
check(s2.MPI_TAG == 5);
check(s2.MPI_ERROR == MPI_ERR_TOPOLOGY);
check(msg == MPI_MESSAGE_NULL);
}
/* test 2: simple send & improbe+mrecv */
if (rank == 0) {
sendbuf[0] = 0xdeadbeef;
sendbuf[1] = 0xfeedface;
MPI_Send(sendbuf, 2, MPI_INT, 1, 5, MPI_COMM_WORLD);
} else {
memset(&s1, 0xab, sizeof(MPI_Status));
memset(&s2, 0xab, sizeof(MPI_Status));
/* the error field should remain unmodified */
s1.MPI_ERROR = MPI_ERR_DIMS;
s2.MPI_ERROR = MPI_ERR_TOPOLOGY;
msg = MPI_MESSAGE_NULL;
do {
check(msg == MPI_MESSAGE_NULL);
MPI_Improbe(0, 5, MPI_COMM_WORLD, &found, &msg, &s1);
} while (!found);
check(msg != MPI_MESSAGE_NULL);
check(s1.MPI_SOURCE == 0);
check(s1.MPI_TAG == 5);
check(s1.MPI_ERROR == MPI_ERR_DIMS);
count = -1;
MPI_Get_count(&s1, MPI_INT, &count);
check(count == 2);
recvbuf[0] = 0x01234567;
recvbuf[1] = 0x89abcdef;
MPI_Mrecv(recvbuf, count, MPI_INT, &msg, &s2);
check(recvbuf[0] == 0xdeadbeef);
check(recvbuf[1] == 0xfeedface);
check(s2.MPI_SOURCE == 0);
check(s2.MPI_TAG == 5);
check(s2.MPI_ERROR == MPI_ERR_TOPOLOGY);
check(msg == MPI_MESSAGE_NULL);
}
/* test 3: simple send & improbe+imrecv */
if (rank == 0) {
sendbuf[0] = 0xdeadbeef;
sendbuf[1] = 0xfeedface;
MPI_Send(sendbuf, 2, MPI_INT, 1, 5, MPI_COMM_WORLD);
} else {
memset(&s1, 0xab, sizeof(MPI_Status));
memset(&s2, 0xab, sizeof(MPI_Status));
/* the error field should remain unmodified */
s1.MPI_ERROR = MPI_ERR_DIMS;
s2.MPI_ERROR = MPI_ERR_TOPOLOGY;
msg = MPI_MESSAGE_NULL;
do {
check(msg == MPI_MESSAGE_NULL);
MPI_Improbe(0, 5, MPI_COMM_WORLD, &found, &msg, &s1);
} while (!found);
check(msg != MPI_MESSAGE_NULL);
check(s1.MPI_SOURCE == 0);
check(s1.MPI_TAG == 5);
check(s1.MPI_ERROR == MPI_ERR_DIMS);
count = -1;
MPI_Get_count(&s1, MPI_INT, &count);
check(count == 2);
rreq = MPI_REQUEST_NULL;
MPI_Imrecv(recvbuf, count, MPI_INT, &msg, &rreq);
check(rreq != MPI_REQUEST_NULL);
MPI_Wait(&rreq, &s2);
check(recvbuf[0] == 0xdeadbeef);
check(recvbuf[1] == 0xfeedface);
check(s2.MPI_SOURCE == 0);
check(s2.MPI_TAG == 5);
check(s2.MPI_ERROR == MPI_ERR_TOPOLOGY);
check(msg == MPI_MESSAGE_NULL);
}
/* test 4: mprobe+mrecv with MPI_PROC_NULL */
{
memset(&s1, 0xab, sizeof(MPI_Status));
memset(&s2, 0xab, sizeof(MPI_Status));
/* the error field should remain unmodified */
s1.MPI_ERROR = MPI_ERR_DIMS;
s2.MPI_ERROR = MPI_ERR_TOPOLOGY;
msg = MPI_MESSAGE_NULL;
MPI_Mprobe(MPI_PROC_NULL, 5, MPI_COMM_WORLD, &msg, &s1);
check(s1.MPI_SOURCE == MPI_PROC_NULL);
check(s1.MPI_TAG == MPI_ANY_TAG);
check(s1.MPI_ERROR == MPI_ERR_DIMS);
check(msg == MPI_MESSAGE_NO_PROC);
count = -1;
MPI_Get_count(&s1, MPI_INT, &count);
check(count == 0);
recvbuf[0] = 0x01234567;
recvbuf[1] = 0x89abcdef;
MPI_Mrecv(recvbuf, count, MPI_INT, &msg, &s2);
/* recvbuf should remain unmodified */
check(recvbuf[0] == 0x01234567);
check(recvbuf[1] == 0x89abcdef);
/* should get back "proc null status" */
check(s2.MPI_SOURCE == MPI_PROC_NULL);
check(s2.MPI_TAG == MPI_ANY_TAG);
check(s2.MPI_ERROR == MPI_ERR_TOPOLOGY);
check(msg == MPI_MESSAGE_NULL);
count = -1;
MPI_Get_count(&s2, MPI_INT, &count);
check(count == 0);
}
/* test 5: mprobe+imrecv with MPI_PROC_NULL */
{
memset(&s1, 0xab, sizeof(MPI_Status));
memset(&s2, 0xab, sizeof(MPI_Status));
/* the error field should remain unmodified */
s1.MPI_ERROR = MPI_ERR_DIMS;
s2.MPI_ERROR = MPI_ERR_TOPOLOGY;
msg = MPI_MESSAGE_NULL;
MPI_Mprobe(MPI_PROC_NULL, 5, MPI_COMM_WORLD, &msg, &s1);
check(s1.MPI_SOURCE == MPI_PROC_NULL);
check(s1.MPI_TAG == MPI_ANY_TAG);
check(s1.MPI_ERROR == MPI_ERR_DIMS);
check(msg == MPI_MESSAGE_NO_PROC);
count = -1;
MPI_Get_count(&s1, MPI_INT, &count);
check(count == 0);
rreq = MPI_REQUEST_NULL;
recvbuf[0] = 0x01234567;
recvbuf[1] = 0x89abcdef;
MPI_Imrecv(recvbuf, count, MPI_INT, &msg, &rreq);
check(rreq != MPI_REQUEST_NULL);
completed = 0;
MPI_Test(&rreq, &completed, &s2); /* single test should always succeed */
check(completed);
/* recvbuf should remain unmodified */
check(recvbuf[0] == 0x01234567);
check(recvbuf[1] == 0x89abcdef);
/* should get back "proc null status" */
check(s2.MPI_SOURCE == MPI_PROC_NULL);
check(s2.MPI_TAG == MPI_ANY_TAG);
check(s2.MPI_ERROR == MPI_ERR_TOPOLOGY);
check(msg == MPI_MESSAGE_NULL);
count = -1;
MPI_Get_count(&s2, MPI_INT, &count);
check(count == 0);
}
/* test 6: improbe+mrecv with MPI_PROC_NULL */
{
memset(&s1, 0xab, sizeof(MPI_Status));
memset(&s2, 0xab, sizeof(MPI_Status));
/* the error field should remain unmodified */
s1.MPI_ERROR = MPI_ERR_DIMS;
s2.MPI_ERROR = MPI_ERR_TOPOLOGY;
msg = MPI_MESSAGE_NULL;
found = 0;
MPI_Improbe(MPI_PROC_NULL, 5, MPI_COMM_WORLD, &found, &msg, &s1);
check(found);
check(msg == MPI_MESSAGE_NO_PROC);
check(s1.MPI_SOURCE == MPI_PROC_NULL);
check(s1.MPI_TAG == MPI_ANY_TAG);
check(s1.MPI_ERROR == MPI_ERR_DIMS);
count = -1;
MPI_Get_count(&s1, MPI_INT, &count);
check(count == 0);
recvbuf[0] = 0x01234567;
recvbuf[1] = 0x89abcdef;
MPI_Mrecv(recvbuf, count, MPI_INT, &msg, &s2);
/* recvbuf should remain unmodified */
check(recvbuf[0] == 0x01234567);
check(recvbuf[1] == 0x89abcdef);
/* should get back "proc null status" */
check(s2.MPI_SOURCE == MPI_PROC_NULL);
check(s2.MPI_TAG == MPI_ANY_TAG);
check(s2.MPI_ERROR == MPI_ERR_TOPOLOGY);
check(msg == MPI_MESSAGE_NULL);
count = -1;
MPI_Get_count(&s2, MPI_INT, &count);
check(count == 0);
}
/* test 7: improbe+imrecv */
{
memset(&s1, 0xab, sizeof(MPI_Status));
memset(&s2, 0xab, sizeof(MPI_Status));
/* the error field should remain unmodified */
s1.MPI_ERROR = MPI_ERR_DIMS;
s2.MPI_ERROR = MPI_ERR_TOPOLOGY;
msg = MPI_MESSAGE_NULL;
MPI_Improbe(MPI_PROC_NULL, 5, MPI_COMM_WORLD, &found, &msg, &s1);
check(found);
check(msg == MPI_MESSAGE_NO_PROC);
check(s1.MPI_SOURCE == MPI_PROC_NULL);
check(s1.MPI_TAG == MPI_ANY_TAG);
check(s1.MPI_ERROR == MPI_ERR_DIMS);
count = -1;
MPI_Get_count(&s1, MPI_INT, &count);
check(count == 0);
rreq = MPI_REQUEST_NULL;
MPI_Imrecv(recvbuf, count, MPI_INT, &msg, &rreq);
check(rreq != MPI_REQUEST_NULL);
completed = 0;
MPI_Test(&rreq, &completed, &s2); /* single test should always succeed */
check(completed);
/* recvbuf should remain unmodified */
check(recvbuf[0] == 0x01234567);
check(recvbuf[1] == 0x89abcdef);
/* should get back "proc null status" */
check(s2.MPI_SOURCE == MPI_PROC_NULL);
check(s2.MPI_TAG == MPI_ANY_TAG);
check(s2.MPI_ERROR == MPI_ERR_TOPOLOGY);
check(msg == MPI_MESSAGE_NULL);
count = -1;
MPI_Get_count(&s2, MPI_INT, &count);
check(count == 0);
}
/* test 8: simple ssend & mprobe+mrecv */
if (rank == 0) {
sendbuf[0] = 0xdeadbeef;
sendbuf[1] = 0xfeedface;
MPI_Ssend(sendbuf, 2, MPI_INT, 1, 5, MPI_COMM_WORLD);
} else {
memset(&s1, 0xab, sizeof(MPI_Status));
memset(&s2, 0xab, sizeof(MPI_Status));
/* the error field should remain unmodified */
s1.MPI_ERROR = MPI_ERR_DIMS;
s2.MPI_ERROR = MPI_ERR_TOPOLOGY;
msg = MPI_MESSAGE_NULL;
MPI_Mprobe(0, 5, MPI_COMM_WORLD, &msg, &s1);
check(s1.MPI_SOURCE == 0);
check(s1.MPI_TAG == 5);
check(s1.MPI_ERROR == MPI_ERR_DIMS);
check(msg != MPI_MESSAGE_NULL);
count = -1;
MPI_Get_count(&s1, MPI_INT, &count);
check(count == 2);
recvbuf[0] = 0x01234567;
recvbuf[1] = 0x89abcdef;
MPI_Mrecv(recvbuf, count, MPI_INT, &msg, &s2);
check(recvbuf[0] == 0xdeadbeef);
check(recvbuf[1] == 0xfeedface);
check(s2.MPI_SOURCE == 0);
check(s2.MPI_TAG == 5);
check(s2.MPI_ERROR == MPI_ERR_TOPOLOGY);
check(msg == MPI_MESSAGE_NULL);
}
/* test 9: mprobe+mrecv LARGE */
if (rank == 0) {
for (i = 0; i < LARGE_SZ; i++)
sendbuf[i] = i;
MPI_Send(sendbuf, LARGE_SZ, MPI_INT, 1, 5, MPI_COMM_WORLD);
} else {
memset(&s1, 0xab, sizeof(MPI_Status));
memset(&s2, 0xab, sizeof(MPI_Status));
/* the error field should remain unmodified */
s1.MPI_ERROR = MPI_ERR_DIMS;
s2.MPI_ERROR = MPI_ERR_TOPOLOGY;
msg = MPI_MESSAGE_NULL;
MPI_Mprobe(0, 5, MPI_COMM_WORLD, &msg, &s1);
check(s1.MPI_SOURCE == 0);
check(s1.MPI_TAG == 5);
check(s1.MPI_ERROR == MPI_ERR_DIMS);
check(msg != MPI_MESSAGE_NULL);
count = -1;
MPI_Get_count(&s1, MPI_INT, &count);
check(count == LARGE_SZ);
memset(recvbuf, 0xFF, LARGE_SZ * sizeof(int));
MPI_Mrecv(recvbuf, count, MPI_INT, &msg, &s2);
for (i = 0; i < LARGE_SZ; i++)
check(recvbuf[i] == i);
check(s2.MPI_SOURCE == 0);
check(s2.MPI_TAG == 5);
check(s2.MPI_ERROR == MPI_ERR_TOPOLOGY);
check(msg == MPI_MESSAGE_NULL);
}
/* test 10: mprobe+mrecv noncontiguous datatype */
MPI_Type_vector(2, 1, 4, MPI_INT, &vectype);
MPI_Type_commit(&vectype);
if (rank == 0) {
memset(sendbuf, 0, 8 * sizeof(int));
sendbuf[0] = 0xdeadbeef;
sendbuf[4] = 0xfeedface;
MPI_Send(sendbuf, 1, vectype, 1, 5, MPI_COMM_WORLD);
} else {
memset(&s1, 0xab, sizeof(MPI_Status));
memset(&s2, 0xab, sizeof(MPI_Status));
/* the error field should remain unmodified */
s1.MPI_ERROR = MPI_ERR_DIMS;
s2.MPI_ERROR = MPI_ERR_TOPOLOGY;
msg = MPI_MESSAGE_NULL;
MPI_Mprobe(0, 5, MPI_COMM_WORLD, &msg, &s1);
check(s1.MPI_SOURCE == 0);
check(s1.MPI_TAG == 5);
check(s1.MPI_ERROR == MPI_ERR_DIMS);
check(msg != MPI_MESSAGE_NULL);
count = -1;
MPI_Get_count(&s1, vectype, &count);
check(count == 1);
memset(recvbuf, 0, 8 * sizeof(int));
MPI_Mrecv(recvbuf, 1, vectype, &msg, &s2);
check(recvbuf[0] == 0xdeadbeef);
for (i = 1; i < 4; i++)
check(recvbuf[i] == 0);
check(recvbuf[4] = 0xfeedface);
for (i = 5; i < 8; i++)
check(recvbuf[i] == 0);
check(s2.MPI_SOURCE == 0);
check(s2.MPI_TAG == 5);
check(s2.MPI_ERROR == MPI_ERR_TOPOLOGY);
check(msg == MPI_MESSAGE_NULL);
}
MPI_Type_free(&vectype);
/* test 11: mprobe+mrecv noncontiguous datatype LARGE */
MPI_Type_vector(LARGE_DIM, LARGE_DIM - 1, LARGE_DIM, MPI_INT, &vectype);
MPI_Type_commit(&vectype);
if (rank == 0) {
for (i = 0; i < LARGE_SZ; i++)
sendbuf[i] = i;
MPI_Send(sendbuf, 1, vectype, 1, 5, MPI_COMM_WORLD);
} else {
int idx = 0;
memset(&s1, 0xab, sizeof(MPI_Status));
memset(&s2, 0xab, sizeof(MPI_Status));
/* the error field should remain unmodified */
s1.MPI_ERROR = MPI_ERR_DIMS;
s2.MPI_ERROR = MPI_ERR_TOPOLOGY;
msg = MPI_MESSAGE_NULL;
MPI_Mprobe(0, 5, MPI_COMM_WORLD, &msg, &s1);
check(s1.MPI_SOURCE == 0);
check(s1.MPI_TAG == 5);
check(s1.MPI_ERROR == MPI_ERR_DIMS);
check(msg != MPI_MESSAGE_NULL);
count = -1;
MPI_Get_count(&s1, vectype, &count);
check(count == 1);
memset(recvbuf, 0, LARGE_SZ * sizeof(int));
MPI_Mrecv(recvbuf, 1, vectype, &msg, &s2);
for (i = 0; i < LARGE_DIM; i++) {
int j;
for (j = 0; j < LARGE_DIM - 1; j++) {
check(recvbuf[idx] == idx);
++idx;
}
check(recvbuf[idx++] == 0);
}
check(s2.MPI_SOURCE == 0);
check(s2.MPI_TAG == 5);
check(s2.MPI_ERROR == MPI_ERR_TOPOLOGY);
check(msg == MPI_MESSAGE_NULL);
}
MPI_Type_free(&vectype);
/* test 12: order test */
if (rank == 0) {
MPI_Request lrequest[2];
sendbuf[0] = 0xdeadbeef;
sendbuf[1] = 0xfeedface;
sendbuf[2] = 0xdeadbeef;
sendbuf[3] = 0xfeedface;
sendbuf[4] = 0xdeadbeef;
sendbuf[5] = 0xfeedface;
MPI_Isend(&sendbuf[0], 4, MPI_INT, 1, 6, MPI_COMM_WORLD, &lrequest[0]);
MPI_Isend(&sendbuf[4], 2, MPI_INT, 1, 6, MPI_COMM_WORLD, &lrequest[1]);
MPI_Waitall(2, &lrequest[0], MPI_STATUSES_IGNORE);
} else {
memset(&s1, 0xab, sizeof(MPI_Status));
memset(&s2, 0xab, sizeof(MPI_Status));
/* the error field should remain unmodified */
s1.MPI_ERROR = MPI_ERR_DIMS;
s2.MPI_ERROR = MPI_ERR_TOPOLOGY;
msg = MPI_MESSAGE_NULL;
MPI_Mprobe(0, 6, MPI_COMM_WORLD, &msg, &s1);
check(s1.MPI_SOURCE == 0);
check(s1.MPI_TAG == 6);
check(s1.MPI_ERROR == MPI_ERR_DIMS);
check(msg != MPI_MESSAGE_NULL);
count = -1;
MPI_Get_count(&s1, MPI_INT, &count);
check(count == 4);
recvbuf[0] = 0x01234567;
recvbuf[1] = 0x89abcdef;
MPI_Recv(recvbuf, 2, MPI_INT, 0, 6, MPI_COMM_WORLD, &s2);
check(s2.MPI_SOURCE == 0);
check(s2.MPI_TAG == 6);
check(recvbuf[0] == 0xdeadbeef);
check(recvbuf[1] == 0xfeedface);
recvbuf[0] = 0x01234567;
recvbuf[1] = 0x89abcdef;
recvbuf[2] = 0x01234567;
recvbuf[3] = 0x89abcdef;
s2.MPI_ERROR = MPI_ERR_TOPOLOGY;
MPI_Mrecv(recvbuf, count, MPI_INT, &msg, &s2);
check(recvbuf[0] == 0xdeadbeef);
check(recvbuf[1] == 0xfeedface);
check(recvbuf[2] == 0xdeadbeef);
check(recvbuf[3] == 0xfeedface);
check(s2.MPI_SOURCE == 0);
check(s2.MPI_TAG == 6);
check(s2.MPI_ERROR == MPI_ERR_TOPOLOGY);
check(msg == MPI_MESSAGE_NULL);
}
/* test 13: simple send & mprobe+mrecv with zero count */
if (rank == 0) {
MPI_Send(sendbuf, 0, MPI_INT, 1, 13, MPI_COMM_WORLD);
} else {
memset(&s1, 0xab, sizeof(MPI_Status));
memset(&s2, 0xab, sizeof(MPI_Status));
/* the error field should remain unmodified */
s1.MPI_ERROR = MPI_ERR_DIMS;
s2.MPI_ERROR = MPI_ERR_TOPOLOGY;
msg = MPI_MESSAGE_NULL;
MPI_Mprobe(0, 13, MPI_COMM_WORLD, &msg, &s1);
check(s1.MPI_SOURCE == 0);
check(s1.MPI_TAG == 13);
check(s1.MPI_ERROR == MPI_ERR_DIMS);
check(msg != MPI_MESSAGE_NULL);
count = -1;
MPI_Get_count(&s1, MPI_INT, &count);
check(count == 0);
recvbuf[0] = 0x01234567;
recvbuf[1] = 0x89abcdef;
MPI_Mrecv(recvbuf, 0, MPI_INT, &msg, &s2);
/* recvbuf should remain unmodified */
check(recvbuf[0] == 0x01234567);
check(recvbuf[1] == 0x89abcdef);
check(s2.MPI_SOURCE == 0);
check(s2.MPI_TAG == 13);
check(s2.MPI_ERROR == MPI_ERR_TOPOLOGY);
check(msg == MPI_MESSAGE_NULL);
count = -1;
MPI_Get_count(&s2, MPI_INT, &count);
check(count == 0);
}
/* test 14: simple send & mprobe+mrecv with zero-size datatype */
if (rank == 0) {
MPI_Send(sendbuf, 0, MPI_BYTE, 1, 14, MPI_COMM_WORLD);
} else {
MPI_Datatype zero_dtype;
MPI_Type_contiguous(0, MPI_INT, &zero_dtype);
MPI_Type_commit(&zero_dtype);
memset(&s1, 0xab, sizeof(MPI_Status));
memset(&s2, 0xab, sizeof(MPI_Status));
/* the error field should remain unmodified */
s1.MPI_ERROR = MPI_ERR_DIMS;
s2.MPI_ERROR = MPI_ERR_TOPOLOGY;
msg = MPI_MESSAGE_NULL;
MPI_Mprobe(0, 14, MPI_COMM_WORLD, &msg, &s1);
check(s1.MPI_SOURCE == 0);
check(s1.MPI_TAG == 14);
check(s1.MPI_ERROR == MPI_ERR_DIMS);
check(msg != MPI_MESSAGE_NULL);
count = -1;
MPI_Get_count(&s1, zero_dtype, &count);
check(count == 0);
recvbuf[0] = 0x01234567;
recvbuf[1] = 0x89abcdef;
MPI_Mrecv(recvbuf, 1, zero_dtype, &msg, &s2);
/* recvbuf should remain unmodified */
check(recvbuf[0] == 0x01234567);
check(recvbuf[1] == 0x89abcdef);
check(s2.MPI_SOURCE == 0);
check(s2.MPI_TAG == 14);
check(s2.MPI_ERROR == MPI_ERR_TOPOLOGY);
check(msg == MPI_MESSAGE_NULL);
count = -1;
MPI_Get_count(&s2, zero_dtype, &count);
check(count == 0);
MPI_Type_free(&zero_dtype);
}
free(sendbuf);
free(recvbuf);
/* TODO MPI_ANY_SOURCE and MPI_ANY_TAG should be tested as well */
/* TODO a full range of message sizes should be tested too */
/* TODO threaded tests are also needed, but they should go in a separate
* program */
/* simple test to ensure that c2f/f2c routines are present (initially missed
* in MPICH impl) */
{
MPI_Fint f_handle = 0xdeadbeef;
f_handle = MPI_Message_c2f(MPI_MESSAGE_NULL);
msg = MPI_Message_f2c(f_handle);
check(f_handle != 0xdeadbeef);
check(msg == MPI_MESSAGE_NULL);
/* PMPI_ versions should also exists */
f_handle = 0xdeadbeef;
f_handle = PMPI_Message_c2f(MPI_MESSAGE_NULL);
msg = PMPI_Message_f2c(f_handle);
check(f_handle != 0xdeadbeef);
check(msg == MPI_MESSAGE_NULL);
}
epilogue:
MTest_Finalize(errs);
return MTestReturnValue(errs);
}
