update store_results

MiloLurati · Jun 17, 2021 · 768bdfb · 768bdfb
1 parent 069d1c6
commit 768bdfb
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/kernel_tuner/integration.py b/kernel_tuner/integration.py
@@ -197,8 +197,11 @@ def top_result(item):
         meta = {}
         meta["version_number"] = "1.0"
         meta["kernel_name"] = kernel_name
-        if kernel_string:
-            meta["kernel_string"] = kernel_string
+        if kernel_string and not callable(kernel_string) and not isinstance(kernel_string, list):
+            if util.looks_like_a_filename(kernel_string):
+                meta["kernel_string"] = util.read_file(kernel_string)
+            else:
+                meta["kernel_string"] = kernel_string
         meta["objective"] = objective
         meta["objective_higher_is_better"] = objective_higher_is_better
         meta["tunable_parameters"] = list(tune_params.keys())
@@ -229,7 +232,7 @@ def top_result(item):
     #write output file
     meta["data"] = data
     with open(results_filename, 'w') as fh:
-        fh.write(json.dumps(meta))
+        fh.write(json.dumps(meta, indent=""))
 
 
 def create_device_targets(header_filename, results_filename, objective=None, objective_higher_is_better=None):