Calculate time spent in LB + heuristic

Author: Quincunx271

util/analyze/lib/compile_times.py

@@ -20,6 +20,22 @@ def sched_time(logs):
     return sum(_block_time(blk) for blk in logs)
 
 
+def heuristic_time_for_blk(blk: Block) -> int:
+    return blk.single('HeuristicResult')['elapsed']
+
+
+def heuristic_time(logs):
+    return sum(heuristic_time_for_blk(b) for b in logs)
+
+
+def first_lower_bound_time_for_blk(blk: Block) -> int:
+    return blk['CostLowerBound'][0]['elapsed']
+
+
+def first_lower_bound_time(logs):
+    return sum(first_lower_bound_time_for_blk(blk) for blk in logs)
+
+
 _CPU2017_TIME_ELAPSED = re.compile(r"Elapsed compile for '(?P<bench>[^']+)': \S+ \((?P<elapsed>\d+)\)")
 _BACKUP_TIME_ELAPSED = re.compile(r'(?P<elapsed>\d+) total seconds elapsed')
 _PLAIDML_TIME_ELAPSED = re.compile(
@@ -30,9 +46,9 @@ def sched_time(logs):
 def shoc_total_compile_time_seconds(logs):
     try:
         elapsed = sum(int(m['elapsed'])
-                   for bench in logs.benchmarks
-                   for blk in bench
-                   for m in _SHOC_TIME_ELAPSED.finditer(blk.raw_log))
+                      for bench in logs.benchmarks
+                      for blk in bench
+                      for m in _SHOC_TIME_ELAPSED.finditer(blk.raw_log))
         return float(elapsed) * 1e-9
     except TypeError:
         raise KeyError('Logs must contain "Finished compiling; total ns = " output by the modified SHOC benchmark suite')

util/gt_analysis/gt_cmp.py

@@ -113,6 +113,10 @@ def compute_stats(nogt: Logs, gt: Logs, *, pass_num: int, total_compile_time_sec
         'Total Sched Time (GT)': sched_time(gt),
         'Enum Time (No GT)': utils.sum_stat_for_all(enum_time_for_blk, nogt),
         'Enum Time (GT)': utils.sum_stat_for_all(enum_time_for_blk, gt),
+        'Lower Bound Time (No GT)': compile_times.first_lower_bound_time(nogt),
+        'Lower Bound Time (GT)': compile_times.first_lower_bound_time(gt),
+        'Heuristic Time (No GT)': compile_times.heuristic_time(nogt),
+        'Heuristic Time (GT)': compile_times.heuristic_time(gt),
         'Total GT Time': utils.sum_stat_for_all(total_gt_elapsed_for_blk, gt),
 
         'Total Sched Time (opt. blks only) (No GT)': sched_time(nogt_opt),