int main(int argc, char **argv) {
eb_t eb;
ep_t ep;
uint64_t networkpid;
int myproc;
int numprocs;
int k;
int iters = 0;
int errs = 0;
TEST_STARTUP(argc, argv, networkpid, eb, ep, 1, 1, "iters");
/* setup handlers */
setupUtilHandlers(ep, eb);
/* get SPMD info */
myproc = AMX_SPMDMyProc();
numprocs = AMX_SPMDNumProcs();
if (argc > 1) iters = atoi(argv[1]);
if (!iters) iters = 1;
if (myproc == 0) {
printf("Running %i iterations of read/write test...\n", iters);
fflush(stdout);
}
/* gather pointers to static data, to handle non-uniform address spaces */
AM_Safe(AMX_SPMDAllGather(&myvals, pvals, sizeof(myvals)));
for (k=0;k < iters; k++) {
/* set left neighbor's array */
int i;
int leftP = myproc-1;
if (leftP == -1) leftP = numprocs-1;
for (i=0;i<MAX_PROCS;i++) writeWord(leftP, pvals[leftP]+i, k);
writeSync();
AM_Safe(AMX_SPMDBarrier()); /* barrier */
{ /* read right neighbor's array */
int i;
int rightP = myproc+1;
if (rightP == numprocs) rightP = 0;
for (i=0;i<MAX_PROCS;i++) readWord(&readarray[i], rightP, pvals[rightP]+i);
readSync();
/* verify */
for (i=0;i<MAX_PROCS;i++) {
if (((int)readarray[i]) != k) {
printf("ERROR: Proc %i READ/WRITE TEST FAILED : readarray[%i] = %i k = %i\n", myproc, i, (int)readarray[i], k);
fflush(stdout);
errs++;
break;
}
}
}
AM_Safe(AMX_SPMDBarrier()); /* barrier */
}
if (!errs) {
printf("Proc %i verified.\n", myproc);
fflush(stdout);
}
/* dump stats */
AM_Safe(AMX_SPMDBarrier());
printGlobalStats();
AM_Safe(AMX_SPMDBarrier());
/* exit */
AM_Safe(AMX_SPMDExit(0));
return 0;
}
/* usage: testlatency numprocs spawnfn iters P/B
*/
int main(int argc, char **argv) {
uint64_t networkpid;
int64_t begin, end, total;
int polling = 1;
int k;
int iters = 0;
TEST_STARTUP(argc, argv, networkpid, eb, ep, 1, 2, "iters (Poll/Block)");
/* setup handlers */
AM_Safe(AM_SetHandler(ep, PING_REQ_HANDLER, ping_request_handler));
AM_Safe(AM_SetHandler(ep, PING_REP_HANDLER, ping_reply_handler));
setupUtilHandlers(ep, eb);
/* get SPMD info */
myproc = AMX_SPMDMyProc();
numprocs = AMX_SPMDNumProcs();
if (argc > 1) iters = atoi(argv[1]);
if (!iters) iters = 1;
if (argc > 2) {
switch(argv[2][0]) {
case 'p': case 'P': polling = 1; break;
case 'b': case 'B': polling = 0; break;
default: printf("polling must be 'P' or 'B'..\n"); AMX_SPMDExit(1);
}
}
outputTimerStats();
AM_Safe(AMX_SPMDBarrier());
if (myproc == 0) printf("Running %i iterations of latency test...\n", iters);
if (myproc == 0 && numprocs > 1) numleft = (numprocs-1)*iters;
AM_Safe(AMX_SPMDBarrier());
begin = getCurrentTimeMicrosec();
if (myproc == 0 && numprocs > 1) {
mywait(polling);
} else { /* everybody sends packets to 0 */
int expect = (numprocs > 1 ? 1 : 2);
for (k=0;k < iters; k++) {
numleft = expect;
#if VERBOSE
printf("%i: sending request...", myproc); fflush(stdout);
#endif
AM_Safe(AM_Request0(ep, 0, PING_REQ_HANDLER));
mywait(polling);
}
}
end = getCurrentTimeMicrosec();
total = end - begin;
if (myproc != 0 || numprocs == 1) printf("Slave %i: %i microseconds total, throughput: %i requests/sec (%.3f us / request)\n",
myproc, (int)total, (int)(((float)1000000)*iters/((int)total)), ((double)total)/iters);
else printf("Slave 0 done.\n");
fflush(stdout);
/* dump stats */
AM_Safe(AMX_SPMDBarrier());
printGlobalStats();
AM_Safe(AMX_SPMDBarrier());
/* exit */
AM_Safe(AMX_SPMDExit(0));
return 0;
}
