contrib/cortex-strings/scripts/plot-align.py

   1 #!/usr/bin/env python
   2
   3 """Plot the performance of different variants of one routine versus alignment.
   4 """
   5
   6 import libplot
   7
   8 import pylab
   9
  10
  11 def plot(records, bytes, function):
  12     records = [x for x in records if x.bytes==bytes and x.function==function]
  13
  14     variants = libplot.unique(records, 'variant', prefer='this')
  15     alignments = libplot.unique(records, ('src_alignment', 'dst_alignment'))
  16
  17     X = pylab.arange(len(alignments))
  18     width = 1.0/(len(variants)+1)
  19
  20     colours = libplot.make_colours()
  21
  22     pylab.figure(1).set_size_inches((16, 12))
  23     pylab.clf()
  24
  25     for i, variant in enumerate(variants):
  26         heights = []
  27
  28         for alignment in alignments:
  29             matches = [x for x in records if x.variant==variant and x.src_alignment==alignment[0] and x.dst_alignment==alignment[1]]
  30
  31             if matches:
  32                 vals = [match.bytes*match.loops/match.elapsed/(1024*1024) for
  33                         match in matches]
  34                 mean = sum(vals)/len(vals)
  35                 heights.append(mean)
  36             else:
  37                 heights.append(0)
  38
  39         pylab.bar(X+i*width, heights, width, color=colours.next(), label=variant)
  40
  41
  42     axes = pylab.axes()
  43     if libplot.alignments_equal(alignments):
  44         alignment_labels = ["%s" % x[0] for x in alignments]
  45     else:
  46         alignment_labels = ["%s:%s" % (x[0], x[1]) for x in alignments]
  47     axes.set_xticklabels(alignment_labels)
  48     axes.set_xticks(X + 0.5)
  49
  50     pylab.title('Performance of different variants of %(function)s for %(bytes)d byte blocks' % locals())
  51     pylab.xlabel('Alignment')
  52     pylab.ylabel('Rate (MB/s)')
  53     pylab.legend(loc='lower right', ncol=3)
  54     pylab.grid()
  55     pylab.savefig('alignment-%(function)s-%(bytes)d.png' % locals(), dpi=72)
  56
  57 def main():
  58     records = libplot.parse()
  59
  60     for function in libplot.unique(records, 'function'):
  61         for bytes in libplot.unique(records, 'bytes'):
  62             plot(records, bytes, function)
  63
  64     pylab.show()
  65
  66 if __name__ == '__main__':
  67     main()