add timing code back

uwescience · Jun 6, 2014 · 2747ca5 · 2747ca5
1 parent e64d5b2
commit 2747ca5
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/raco/grappalang.py b/raco/grappalang.py
@@ -89,7 +89,7 @@ def group_wrap(ident, grpcode, attrs):
         auto end_%(ident)s = walltime();
         auto runtime_%(ident)s = end_%(ident)s - start_%(ident)s;
         %(timer_metric)s += runtime_%(ident)s;
-        VLOG(1) << "pipeline %(ident)s: " << runtime_%(ident)s << " s";
+        VLOG(1) << "pipeline group %(ident)s: " << runtime_%(ident)s << " s";
         """)
 
         timer_metric = None
@@ -105,6 +105,17 @@ def group_wrap(ident, grpcode, attrs):
     def pipeline_wrap(ident, plcode, attrs):
         code = plcode
 
+        # timing code
+        if True:
+            inner_code = code
+            timing_template = ct("""auto start_%(ident)s = walltime();
+            %(inner_code)s
+            auto end_%(ident)s = walltime();
+            auto runtime_%(ident)s = end_%(ident)s - start_%(ident)s;
+            VLOG(1) << "pipeline %(ident)s: " << runtime_%(ident)s << " s";
+            """)
+            code = timing_template % locals()
+
         syncname = attrs.get('sync')
         if syncname:
             inner_code = code