| Index: bin/compare
|
| diff --git a/bin/compare b/bin/compare
|
| index f723c083c1a0ff89097ce0d77e0cb82c306f0788..82f85d5fd91ab724e5c3aa6e2f1e65508fa10f79 100755
|
| --- a/bin/compare
|
| +++ b/bin/compare
|
| @@ -1,12 +1,24 @@
|
| #!/usr/bin/env python
|
|
|
| +import argparse
|
| +import numpy
|
| import sys
|
| from scipy.stats import mannwhitneyu
|
| +from scipy.stats import sem
|
|
|
| SIGNIFICANCE_THRESHOLD = 0.0001
|
|
|
| +parser = argparse.ArgumentParser(
|
| + formatter_class=argparse.RawDescriptionHelpFormatter,
|
| + description='Compare performance of two runs from nanobench.')
|
| +parser.add_argument('--use_means', action='store_true', default=False,
|
| + help='Use means to calculate performance ratios.')
|
| +parser.add_argument('baseline', help='Baseline file.')
|
| +parser.add_argument('experiment', help='Experiment file.')
|
| +args = parser.parse_args()
|
| +
|
| a,b = {},{}
|
| -for (path, d) in [(sys.argv[1], a), (sys.argv[2], b)]:
|
| +for (path, d) in [(args.baseline, a), (args.experiment, b)]:
|
| for line in open(path):
|
| try:
|
| tokens = line.split()
|
| @@ -23,8 +35,13 @@ common = set(a.keys()).intersection(b.keys())
|
| ps = []
|
| for key in common:
|
| _, p = mannwhitneyu(a[key], b[key]) # Non-parametric t-test. Doesn't assume normal dist.
|
| - am, bm = min(a[key]), min(b[key])
|
| - ps.append((bm/am, p, key, am, bm))
|
| + if args.use_means:
|
| + am, bm = numpy.mean(a[key]), numpy.mean(b[key])
|
| + asem, bsem = sem(a[key]), sem(b[key])
|
| + else:
|
| + am, bm = min(a[key]), min(b[key])
|
| + asem, bsem = 0, 0
|
| + ps.append((bm/am, p, key, am, bm, asem, bsem))
|
| ps.sort(reverse=True)
|
|
|
| def humanize(ns):
|
| @@ -36,7 +53,11 @@ maxlen = max(map(len, common))
|
|
|
| # We print only signficant changes in benchmark timing distribution.
|
| bonferroni = SIGNIFICANCE_THRESHOLD / len(ps) # Adjust for the fact we've run multiple tests.
|
| -for ratio, p, key, am, bm in ps:
|
| +for ratio, p, key, am, bm, asem, bsem in ps:
|
| if p < bonferroni:
|
| str_ratio = ('%.2gx' if ratio < 1 else '%.3gx') % ratio
|
| - print '%*s\t%6s -> %6s\t%s' % (maxlen, key, humanize(am), humanize(bm), str_ratio)
|
| + if args.use_means:
|
| + print '%*s\t%6s(%6s) -> %6s(%6s)\t%s' % (maxlen, key, humanize(am), humanize(asem),
|
| + humanize(bm), humanize(bsem), str_ratio)
|
| + else:
|
| + print '%*s\t%6s -> %6s\t%s' % (maxlen, key, humanize(am), humanize(bm), str_ratio)
|
|
|