M-Labs
/
artiq
mirror of https://github.com/m-labs/artiq.git
2
0
Fork 2
Code Issues Projects Releases Wiki Activity

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
artiq/compiler/testbench/perf.py
View File

@ -1,4 +1,4 @@
import sys, os, time import sys, os, time, cProfile as profile, pstats
from pythonparser import diagnostic from pythonparser import diagnostic
from .. import Module from .. import Module
from ..targets import OR1KTarget from ..targets import OR1KTarget
@ -21,6 +21,9 @@ def main():
for filename in sys.argv[1:]] for filename in sys.argv[1:]]
def benchmark(f, name): def benchmark(f, name):
profiler = profile.Profile()
profiler.enable()
start = time.perf_counter() start = time.perf_counter()
end = 0 end = 0
runs = 0 runs = 0
@ -29,9 +32,14 @@ def main():
runs += 1 runs += 1
end = time.perf_counter() end = time.perf_counter()
profiler.create_stats()
print("{} {} runs: {:.2f}s, {:.2f}ms/run".format( print("{} {} runs: {:.2f}s, {:.2f}ms/run".format(
runs, name, end - start, (end - start) / runs * 1000)) runs, name, end - start, (end - start) / runs * 1000))
stats = pstats.Stats(profiler)
stats.strip_dirs().sort_stats('time').print_stats(10)
sources = [] sources = []
for filename in sys.argv[1:]: for filename in sys.argv[1:]:
with open(filename) as f: with open(filename) as f: