void
chksum_test(int iters)
{
int i;
int iamsender, iamreceiver;
int received;
#ifdef VERBOSE
int nloop = 0;
#endif
iamsender = (myproc % 2 == 0);
iamreceiver = !iamsender;
BARRIER();
if (iamsender) {
for (i = 0; i < iters; i++)
GASNET_Safe(
gasnet_AMRequestShort2((gasnet_node_t)peerproc,
201, i, _mseed[i].seed));
}
while ( (received = gasnett_atomic_read(&chksum_received,0)) < iters ) {
/*
if (iamreceiver) {
if (received % 5 == 0) {
printf("sleep 1\n");
sleep(1);
}
}
*/
#ifdef VERBOSE
nloop++;
if (nloop % 1000 == 0) {
printf("TEST[%d] nloop = %d chksum_received = %d\n",
myproc,nloop,received);
}
#endif
gasnet_AMPoll();
}
#ifdef VERBOSE
printf("TEST[%d] COMPLETE: nloop = %d chksum_received = %d\n",
myproc,nloop,received);
#endif
BARRIER();
if (iamsender) {
int success = gasnett_atomic_read(&chksum_success,0);
printf("chksum_test(%d) passed %d/%d\n", chksum_iters,
success, received);
}
}
void test_mpi(threaddata_t *tdata) {
MPI_Request sendhandle = MPI_REQUEST_NULL;
MPI_Request recvhandle = MPI_REQUEST_NULL;
int peer = tdata->tid_peer;
int node = tt_thread_map[peer];
int mpipeer = gasnetnode_to_mpirank[node];
int sz;
char *sendbuf;
char *recvbuf;
int tag = tdata->tid;
int i;
do { sz = RANDOM_SIZE(); } while (sz == 0); /* some MPI's may barf on 0 byte send/recv */
sendbuf = (char*)test_malloc(sz);
recvbuf = (char*)test_malloc(sz);
for (i=0; i < MIN(sz,4096); i++) { /* randomize at least the first 4 KB */
sendbuf[i] = (char)rand();
}
ACTION_PRINTF("tid=%3d> starting MPI ping-pong with tid=%3d.\n", tdata->tid, peer);
MPI_LOCK();
ACTION_PRINTF("tid=%3d> setting MPI_Irecv, %i bytes\n", tdata->tid, sz);
MPI_SAFE(MPI_Irecv(recvbuf, sz, MPI_BYTE, mpipeer, 10000+tag, MPI_COMM_WORLD, &recvhandle));
assert(recvhandle != MPI_REQUEST_NULL);
ACTION_PRINTF("tid=%3d> sending MPI message, %i bytes\n", tdata->tid, sz);
MPI_SAFE(MPI_Isend(sendbuf, sz, MPI_BYTE, mpipeer, tag, MPI_COMM_WORLD, &sendhandle));
assert(sendhandle != MPI_REQUEST_NULL);
MPI_UNLOCK();
tdata->flag = -1;
gasnett_local_wmb();
ACTION_PRINTF("tid=%3d> MPI AMShortRequest to tid=%3d\n", tdata->tid, peer);
GASNET_Safe(gasnet_AMRequestShort2(node, hidx_mpi_handler, tdata->tid, sz));
while (tdata->flag != 0) {
ACTION_PRINTF("tid=%3d> MPI probe AMShortRequest to tid=%3d\n", tdata->tid, peer);
GASNET_Safe(gasnet_AMRequestShort1(node, hidx_mpi_probehandler, tdata->tid));
gasnett_sched_yield();
test_sleep(tdata);
GASNET_Safe(gasnet_AMPoll());
mpi_test(&sendhandle); /* occasional testing may be required for progress */
mpi_test(&recvhandle);
}
tdata->flag = -1;
mpi_complete(&sendhandle);
mpi_complete(&recvhandle);
/* verify */
for (i=0; i < sz; i++) {
if (sendbuf[i] != recvbuf[i])
FATALERR("mismatch at element %i in MPI test.", i);
}
test_free(sendbuf);
test_free(recvbuf);
ACTION_PRINTF("tid=%3d> MPI ping-pong with tid=%3d complete.\n", tdata->tid, peer);
}
