68 lines
2.0 KiB
Python
68 lines
2.0 KiB
Python
#!/usr/bin/env python
|
|
|
|
"""Plot the performance of different variants of one routine versus alignment.
|
|
"""
|
|
|
|
import libplot
|
|
|
|
import pylab
|
|
|
|
|
|
def plot(records, bytes, function):
|
|
records = [x for x in records if x.bytes==bytes and x.function==function]
|
|
|
|
variants = libplot.unique(records, 'variant', prefer='this')
|
|
alignments = libplot.unique(records, ('src_alignment', 'dst_alignment'))
|
|
|
|
X = pylab.arange(len(alignments))
|
|
width = 1.0/(len(variants)+1)
|
|
|
|
colours = libplot.make_colours()
|
|
|
|
pylab.figure(1).set_size_inches((16, 12))
|
|
pylab.clf()
|
|
|
|
for i, variant in enumerate(variants):
|
|
heights = []
|
|
|
|
for alignment in alignments:
|
|
matches = [x for x in records if x.variant==variant and x.src_alignment==alignment[0] and x.dst_alignment==alignment[1]]
|
|
|
|
if matches:
|
|
vals = [match.bytes*match.loops/match.elapsed/(1024*1024) for
|
|
match in matches]
|
|
mean = sum(vals)/len(vals)
|
|
heights.append(mean)
|
|
else:
|
|
heights.append(0)
|
|
|
|
pylab.bar(X+i*width, heights, width, color=colours.next(), label=variant)
|
|
|
|
|
|
axes = pylab.axes()
|
|
if libplot.alignments_equal(alignments):
|
|
alignment_labels = ["%s" % x[0] for x in alignments]
|
|
else:
|
|
alignment_labels = ["%s:%s" % (x[0], x[1]) for x in alignments]
|
|
axes.set_xticklabels(alignment_labels)
|
|
axes.set_xticks(X + 0.5)
|
|
|
|
pylab.title('Performance of different variants of %(function)s for %(bytes)d byte blocks' % locals())
|
|
pylab.xlabel('Alignment')
|
|
pylab.ylabel('Rate (MB/s)')
|
|
pylab.legend(loc='lower right', ncol=3)
|
|
pylab.grid()
|
|
pylab.savefig('alignment-%(function)s-%(bytes)d.png' % locals(), dpi=72)
|
|
|
|
def main():
|
|
records = libplot.parse()
|
|
|
|
for function in libplot.unique(records, 'function'):
|
|
for bytes in libplot.unique(records, 'bytes'):
|
|
plot(records, bytes, function)
|
|
|
|
pylab.show()
|
|
|
|
if __name__ == '__main__':
|
|
main()
|