Add profiling to the performance testbench.

whitequark · whitequark · commit b0185f3917c9 · 2015-07-29T22:23:22.000+03:00
diff --git a/artiq/compiler/testbench/perf.py b/artiq/compiler/testbench/perf.py
@@ -1,4 +1,4 @@
-import sys, os, time
+import sys, os, time, cProfile as profile, pstats
 from pythonparser import diagnostic
 from .. import Module
 from ..targets import OR1KTarget
@@ -21,6 +21,9 @@ def process_diagnostic(diag):
                for filename in sys.argv[1:]]
 
     def benchmark(f, name):
+        profiler = profile.Profile()
+        profiler.enable()
+
         start = time.perf_counter()
         end   = 0
         runs  = 0
@@ -29,9 +32,14 @@ def benchmark(f, name):
             runs += 1
             end = time.perf_counter()
 
+        profiler.create_stats()
+
         print("{} {} runs: {:.2f}s, {:.2f}ms/run".format(
                 runs, name, end - start, (end - start) / runs * 1000))
 
+        stats = pstats.Stats(profiler)
+        stats.strip_dirs().sort_stats('time').print_stats(10)
+
     sources = []
     for filename in sys.argv[1:]:
         with open(filename) as f:
