Merge pull request #419 from CNugteren/fix_tuner_out_of_bounds_access

Fix tuner printing issue

1 files changed, 12 insertions, 2 deletions

diff --git a/src/tuning/tuning.cpp b/src/tuning/tuning.cpp
index 4160faf7..36da4a80 100644
--- a/src/tuning/tuning.cpp
+++ b/src/tuning/tuning.cpp
@@ -227,7 +227,12 @@ void Tuner(int argc, char* argv[], const int V,
     for (auto i=size_t{0}; i<global.size(); ++i) {
       while ((global[i] / local[i]) * local[i] != global[i]) { global[i]++; }
     }
-    printf("%8zu%8zu |%8zu%8zu |", local[0], local[1], global[0], global[1]);
+    if (local.size() > 1 && global.size() > 1) {
+      printf("%8zu%8zu |%8zu%8zu |", local[0], local[1], global[0], global[1]);
+    }
+    else {
+      printf("%8zu%8d |%8zu%8d |", local[0], 1, global[0], 1);
+    }
 
     // Compiles the kernel
     auto compiler_options = std::vector<std::string>();
@@ -283,7 +288,12 @@ void Tuner(int argc, char* argv[], const int V,
       for (auto i=size_t{0}; i<global.size(); ++i) {
         while ((global[i] / local[i]) * local[i] != global[i]) { global[i]++; }
       }
-      printf("%8zu%8zu |%8zu%8zu |", local[0], local[1], global[0], global[1]);
+      if (local.size() > 1 && global.size() > 1) {
+        printf("%8zu%8zu |%8zu%8zu |", local[0], local[1], global[0], global[1]);
+      }
+      else {
+        printf("%8zu%8d |%8zu%8d |", local[0], 1, global[0], 1);
+      }
 
       // Sets the parameters for this configuration
       auto kernel_source = std::string{""};