LoopTree latency analysis bugfix

Author: Michael Gilbert

pytimeloop/isl/singular.py

@@ -4,6 +4,7 @@
 
 
 def get_sum_of_pw_qpolynomial(pw_qp):
+    """Get the sum of pw_qp over all points in the domain."""
     if isinstance(pw_qp, Number):
         return pw_qp
 

pytimeloop/looptree/latency/processors.py

@@ -1,13 +1,23 @@
+import pprint
+
 import islpy as isl
 
 from bindings.looptree import PipelineSpatialTag
 from pytimeloop.isl.sum import sum_until_idx, make_reduction_map
 from pytimeloop.isl.qpolynomial import from_pw_qpolynomial_fold
 
 def process_sequential_latency(top_idx: int, latencies):
-    dim_tags = latencies[0][0]
-    summed_latency = sum(map(lambda pair: pair[1], latencies))
-    return dim_tags[:top_idx], sum_until_idx(top_idx, summed_latency)
+    common_dim_tags = latencies[0][0][:top_idx]
+    try:
+        total_sequential_latency = sum(
+            sum_until_idx(top_idx, latency)
+            for dim_tags, latency in latencies
+        )
+    except:
+        print('Bad input:')
+        pprint.pp(latencies)
+        raise
+    return common_dim_tags, total_sequential_latency
 
 
 def process_pipeline_latency(top_idx: int, latencies):