Author: Carl Friedrich Bolz <[email protected]>
Branch: extradoc
Changeset: r4207:dc054d70476d
Date: 2012-05-14 14:03 +0200
http://bitbucket.org/pypy/extradoc/changeset/dc054d70476d/
Log: a script that I had never checked in
diff --git a/talk/pepm2011/benchmarks/opt-diff.py
b/talk/pepm2011/benchmarks/opt-diff.py
new file mode 100755
--- /dev/null
+++ b/talk/pepm2011/benchmarks/opt-diff.py
@@ -0,0 +1,118 @@
+#!/usr/bin/env python
+"""
+Parse and display the traces produced by pypy-c-jit when PYPYLOG is set.
+"""
+
+import py
+import sys
+import optparse
+from pprint import pprint
+from pypy.tool import logparser
+from pypy.jit.metainterp.test.oparser import parse
+from pypy.jit.metainterp.history import ConstInt
+from pypy.rpython.lltypesystem import llmemory, lltype
+
+categories = {
+ 'setfield_gc': 'set',
+ 'setarrayitem_gc': 'set',
+ 'strsetitem': 'set',
+ 'getfield_gc': 'get',
+ 'getfield_gc_pure': 'get',
+ 'getarrayitem_gc': 'get',
+ 'getarrayitem_gc_pure': 'get',
+ 'strgetitem': 'get',
+ 'new': 'new',
+ 'new_array': 'new',
+ 'newstr': 'new',
+ 'new_with_vtable': 'new',
+ 'virtual_ref': 'new',
+ 'guard_class': 'guard',
+ 'guard_nonnull_class': 'guard',
+}
+
+def summarize(self, adding_insns={}): # for debugging
+ insns = adding_insns.copy()
+ for op in self._all_operations(omit_finish=True):
+ opname = op.getopname()
+ opname = categories.get(opname, 'rest')
+ insns[opname] = insns.get(opname, 0) + 1
+ return insns
+
+def main(loopfile, options):
+ log = logparser.parse_log_file(loopfile)
+ loops, summary = consider_category(log, options, "jit-log-opt-")
+ if not options.quiet:
+ for loop in loops:
+ loop.show()
+
+ if options.summary:
+ print
+ print 'Summary:'
+ print_summary(summary)
+
+ if options.diff:
+ # non-optimized loops and summary
+ nloops, nsummary = consider_category(log, options, "jit-log-noopt-")
+ diff = {}
+ keys = set(summary.keys()).union(set(nsummary))
+ for key in keys:
+ before = nsummary.get(key, 0)
+ after = summary.get(key, 0)
+ diff[key] = (before-after, before, after)
+ print_diff(diff)
+
+def consider_category(log, options, category):
+ loops = logparser.extract_category(log, category)
+ if options.loopnum is None:
+ input_loops = loops
+ else:
+ input_loops = [loops[options.loopnum]]
+ loops = [parse(inp, no_namespace=True, nonstrict=True)
+ for inp in input_loops]
+ summary = {}
+ for loop in loops:
+ summary = summarize(loop, summary)
+ return loops, summary
+
+
+def print_summary(summary):
+ ops = [(summary[key], key) for key in summary]
+ ops.sort(reverse=True)
+ for n, key in ops:
+ print '%5d' % n, key
+
+def print_diff(diff):
+ ops = [(key, before, after, d) for key, (d, before, after) in
diff.iteritems()]
+ ops.sort(reverse=True)
+ tot_before = 0
+ tot_after = 0
+ print ",",
+ for key, before, after, d in ops:
+ print key, ", ,",
+ print "total"
+ print args[0], ",",
+ for key, before, after, d in ops:
+ tot_before += before
+ tot_after += after
+ print before, ",", after, ",",
+ print tot_before, ",", tot_after
+
+if __name__ == '__main__':
+ parser = optparse.OptionParser(usage="%prog loopfile [options]")
+ parser.add_option('-n', '--loopnum', dest='loopnum', default=-1,
metavar='N', type=int,
+ help='show the loop number N [default: last]')
+ parser.add_option('-a', '--all', dest='loopnum', action='store_const',
const=None,
+ help='show all loops in the file')
+ parser.add_option('-s', '--summary', dest='summary', action='store_true',
default=False,
+ help='print a summary of the operations in the loop(s)')
+ parser.add_option('-d', '--diff', dest='diff', action='store_true',
default=False,
+ help='print the difference between non-optimized and
optimized operations in the loop(s)')
+ parser.add_option('-q', '--quiet', dest='quiet', action='store_true',
default=False,
+ help='do not show the graphical representation of the
loop')
+
+ options, args = parser.parse_args()
+ if len(args) != 1:
+ parser.print_help()
+ sys.exit(2)
+
+ main(args[0], options)
_______________________________________________
pypy-commit mailing list
[email protected]
http://mail.python.org/mailman/listinfo/pypy-commit
