Skip to content

Commit 7a490c8

Browse files
committed
Backmerged #281 to fix iteration number
1 parent ca87879 commit 7a490c8

File tree

1 file changed

+8
-3
lines changed

1 file changed

+8
-3
lines changed

src/mdtest.c

Lines changed: 8 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1372,6 +1372,11 @@ void print_help (void) {
13721372
exit(0);
13731373
}
13741374

1375+
int calc_allreduce_index(int iter, int rank, int op){
1376+
int tableSize = MDTEST_LAST_NUM;
1377+
return iter * tableSize * size + rank * tableSize + op;
1378+
}
1379+
13751380
void summarize_results(int iterations) {
13761381
char access[MAX_PATHLEN];
13771382
int i, j, k;
@@ -1436,7 +1441,8 @@ void summarize_results(int iterations) {
14361441
for (j=0; j<iterations; j++) {
14371442
maxes[j] = all[j*tableSize + i];
14381443
for (k=0; k<size; k++) {
1439-
curr = all[(k*tableSize*iterations) + (j*tableSize) + i];
1444+
curr = all[calc_allreduce_index(j, k, i)];
1445+
//For verification in 3.2 printf("%d %d %d = %f\n", j, k, i, curr);
14401446
if (maxes[j] < curr) {
14411447
maxes[j] = curr;
14421448
}
@@ -1488,8 +1494,7 @@ void summarize_results(int iterations) {
14881494
min = max = all[i];
14891495
for (k=0; k < size; k++) {
14901496
for (j = 0; j < iterations; j++) {
1491-
curr = all[(k*tableSize*iterations)
1492-
+ (j*tableSize) + i];
1497+
curr = all[calc_allreduce_index(j, k, i)];
14931498
if (min > curr) {
14941499
min = curr;
14951500
}

0 commit comments

Comments
 (0)