From 11c784c8bd25292001d27ca0858d2086df562419 Mon Sep 17 00:00:00 2001
From: "Julian M. Kunkel" <juliankunkel@googlemail.com>
Date: Sat, 28 Nov 2020 10:40:41 +0000
Subject: [PATCH] Integrate review feedback.

---
 src/mdtest.c | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/src/mdtest.c b/src/mdtest.c
index 145fea1..ece7ab0 100644
--- a/src/mdtest.c
+++ b/src/mdtest.c
@@ -1271,6 +1271,11 @@ char const * mdtest_test_name(int i){
   return NULL;
 }
 
+int calc_allreduce_index(int iter, int rank, int op){
+  int tableSize = MDTEST_LAST_NUM;
+  return iter*tableSize*size + rank * tableSize + op;
+}
+
 void summarize_results(int iterations, int print_time) {
     char const * access;
     int i, j, k;
@@ -1339,7 +1344,7 @@ void summarize_results(int iterations, int print_time) {
           }
           fprintf(out_logfile, "Test %s", access);
           for (k=0; k < size; k++) {
-            curr = all[j*tableSize*size + k * tableSize + i];
+            curr = all[calc_allreduce_index(j, k, i)];
             fprintf(out_logfile, "%c%e", (k==0 ? ' ': ','), curr);
           }
           fprintf(out_logfile, "\n");
@@ -1355,7 +1360,7 @@ void summarize_results(int iterations, int print_time) {
             min = max = all[i];
             for (j = 0; j < iterations; j++) {
                 for (k=0; k < size; k++) {
-                    curr = all[j*tableSize*size + k*tableSize + i];
+                    curr = all[calc_allreduce_index(j, k, i)];
                     if (min > curr) {
                         min = curr;
                     }