Skip to content

Commit 4ddef4c

Browse files
committed
Moved 'times' from measurements to runtimes to be compatible with schema
1 parent 42a4907 commit 4ddef4c

File tree

2 files changed

+3
-2
lines changed

2 files changed

+3
-2
lines changed

kernel_tuner/file_utils.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@ def store_output_file(output_filename, results, tune_params, objective="time"):
2626
output_filename += ".json"
2727

2828
timing_keys = ["compile_time", "benchmark_time", "framework_time", "strategy_time", "verification_time"]
29-
not_measurement_keys = list(tune_params.keys()) + timing_keys + ["timestamp"]
29+
not_measurement_keys = list(tune_params.keys()) + timing_keys + ["timestamp"] + ["times"]
3030

3131
output_data = []
3232

@@ -45,6 +45,7 @@ def store_output_file(output_filename, results, tune_params, objective="time"):
4545
timings["framework"] = result["framework_time"]
4646
timings["search_algorithm"] = result["strategy_time"]
4747
timings["validation"] = result["verification_time"]
48+
timings["runtimes"] = result["times"]
4849
out["times"] = timings
4950

5051
# encode the validity of the configuration

test/test_cache_file.json

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -310,4 +310,4 @@
310310
"timestamp": "2022-12-23 12:11:26.411558+00:00"
311311
}
312312
}
313-
}
313+
}

0 commit comments

Comments
 (0)