Added support for the convgemm tuner in the tuner database

2 files changed, 9 insertions, 1 deletions

diff --git a/scripts/database/database/clblast.py b/scripts/database/database/clblast.py
index 86444f68..ce76f305 100644
--- a/scripts/database/database/clblast.py
+++ b/scripts/database/database/clblast.py
@@ -24,7 +24,9 @@ DEVICE_ATTRIBUTES = ["clblast_device_name", "clblast_device_architecture",
                      "device_core_clock", "device_compute_units"]
 KERNEL_ATTRIBUTES = ["precision", "kernel_family"]
 ARGUMENT_ATTRIBUTES = ["arg_m", "arg_n", "arg_k", "arg_alpha", "arg_beta",
-                       "arg_from", "arg_to", "arg_step"]
+                       "arg_from", "arg_to", "arg_step",
+                       "arg_channels", "arg_height", "arg_width", "arg_kernel_h", "arg_kernel_w",
+                       "arg_num_kernels", "arg_batch_count"]
 ATTRIBUTES = DEVICE_ATTRIBUTES + DEVICE_TYPE_ATTRIBUTES + KERNEL_ATTRIBUTES + ARGUMENT_ATTRIBUTES
 GROUP_ATTRIBUTES = DEVICE_TYPE_ATTRIBUTES + KERNEL_ATTRIBUTES + ["kernel"] + ARGUMENT_ATTRIBUTES
 
diff --git a/src/tuning/tuning.cpp b/src/tuning/tuning.cpp
index f76af774..c5eee527 100644
--- a/src/tuning/tuning.cpp
+++ b/src/tuning/tuning.cpp
@@ -389,6 +389,12 @@ void Tuner(int argc, char* argv[], const int V,
     if (o == kArgAlpha) { metadata.push_back({"arg_alpha", ToString(args.alpha)}); }
     if (o == kArgBeta)  { metadata.push_back({"arg_beta", ToString(args.beta)}); }
     if (o == kArgBatchCount) { metadata.push_back({"arg_batch_count", ToString(args.batch_count)}); }
+    if (o == kArgHeight)     { metadata.push_back({"arg_height", ToString(args.height)}); }
+    if (o == kArgWidth)      { metadata.push_back({"arg_width", ToString(args.width)}); }
+    if (o == kArgKernelH)    { metadata.push_back({"arg_kernel_h", ToString(args.kernel_h)}); }
+    if (o == kArgKernelW)    { metadata.push_back({"arg_kernel_w", ToString(args.kernel_w)}); }
+    if (o == kArgChannels)   { metadata.push_back({"arg_channels", ToString(args.channels)}); }
+    if (o == kArgNumKernels) { metadata.push_back({"arg_num_kernels", ToString(args.num_kernels)}); }
   }
   PrintTimingsToFileAsJSON("clblast_" + settings.kernel_family + "_" + precision_string + ".json",
                            device, platform, metadata, results);