Added collecting and printing of scores for the kernel-selection tuner

1 files changed, 34 insertions, 3 deletions

diff --git a/src/utilities/timing.hpp b/src/utilities/timing.hpp
index 4622aa99..423e6e2b 100644
--- a/src/utilities/timing.hpp
+++ b/src/utilities/timing.hpp
@@ -65,13 +65,18 @@ std::vector<Timing> TimeRoutine(const size_t from, const size_t to, const size_t
 
 // =================================================================================================
 
+using TuningParameter = std::pair<std::string, size_t>;
+using TuningParameters = std::vector<TuningParameter>;
+struct TuningResult { std::string name; double score; TuningParameters parameters; };
+
 void PrintTimingsToFileAsJSON(const std::string &filename,
                               const Device& device, const Platform& platform,
-                              const std::vector<std::pair<std::string,std::string>> &descriptions) {
+                              const std::vector<std::pair<std::string,std::string>> &metadata,
+                              const std::vector<TuningResult>& tuning_results) {
   auto file = fopen(filename.c_str(), "w");
   fprintf(file, "{\n");
-  for (auto &description: descriptions) {
-    fprintf(file, "  \"%s\": \"%s\",\n", description.first.c_str(), description.second.c_str());
+  for (auto &datum: metadata) {
+    fprintf(file, "  \"%s\": \"%s\",\n", datum.first.c_str(), datum.second.c_str());
   }
   fprintf(file, "  \"platform_version\": \"%s\",\n", platform.Version().c_str());
   fprintf(file, "  \"device_name\": \"%s\",\n", GetDeviceName(device).c_str());
@@ -80,6 +85,32 @@ void PrintTimingsToFileAsJSON(const std::string &filename,
   fprintf(file, "  \"device_architecture\": \"%s\",\n", GetDeviceArchitecture(device).c_str());
   fprintf(file, "  \"device_core_clock\": \"%zu\",\n", device.CoreClock());
   fprintf(file, "  \"device_compute_units\": \"%zu\",\n", device.ComputeUnits());
+  fprintf(file, "  \"results\": [\n");
+
+  // Loops over all results
+  auto num_results = tuning_results.size();
+  for (auto r = size_t{0}; r < num_results; ++r) {
+    auto result = tuning_results[r];
+    fprintf(file, "    {\n");
+    fprintf(file, "      \"kernel\": \"%s\",\n", result.name.c_str());
+    fprintf(file, "      \"time\": %.3lf,\n", result.score);
+
+    // Loops over all the parameters for this result
+    fprintf(file, "      \"parameters\": {");
+    auto num_configs = result.parameters.size();
+    for (auto p=size_t{0}; p<num_configs; ++p) {
+      auto config = result.parameters[p];
+      fprintf(file, "\"%s\": %zu", config.first.c_str(), config.second);
+      if (p < num_configs-1) { fprintf(file, ","); }
+    }
+    fprintf(file, "}\n");
+
+    // The footer
+    fprintf(file, "    }");
+    if (r < num_results - 1) { fprintf(file, ","); }
+    fprintf(file, "\n");
+  }
+  fprintf(file, "  ]\n");
   fprintf(file, "}\n");
   fclose(file);
 }