forked from M-Labs/artiq
1
0
Fork 0

Add profiling to the performance testbench.

This commit is contained in:
whitequark 2015-07-29 22:23:22 +03:00
parent d7f9af4bb5
commit b0185f3917
1 changed files with 9 additions and 1 deletions

View File

@ -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 main():
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 main():
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: