summaryrefslogtreecommitdiff
path: root/util/dram_sweep_plot.py
diff options
context:
space:
mode:
Diffstat (limited to 'util/dram_sweep_plot.py')
-rwxr-xr-xutil/dram_sweep_plot.py151
1 files changed, 151 insertions, 0 deletions
diff --git a/util/dram_sweep_plot.py b/util/dram_sweep_plot.py
new file mode 100755
index 000000000..a57de3d1c
--- /dev/null
+++ b/util/dram_sweep_plot.py
@@ -0,0 +1,151 @@
+#!/usr/bin/env python
+
+# Copyright (c) 2014 ARM Limited
+# All rights reserved
+#
+# The license below extends only to copyright in the software and shall
+# not be construed as granting a license to any other intellectual
+# property including but not limited to intellectual property relating
+# to a hardware implementation of the functionality of the software
+# licensed hereunder. You may use the software subject to the license
+# terms below provided that you ensure that this notice is replicated
+# unmodified and in its entirety in all distributions of the software,
+# modified or unmodified, in source code or in binary form.
+#
+# Redistribution and use in source and binary forms, with or without
+# modification, are permitted provided that the following conditions are
+# met: redistributions of source code must retain the above copyright
+# notice, this list of conditions and the following disclaimer;
+# redistributions in binary form must reproduce the above copyright
+# notice, this list of conditions and the following disclaimer in the
+# documentation and/or other materials provided with the distribution;
+# neither the name of the copyright holders nor the names of its
+# contributors may be used to endorse or promote products derived from
+# this software without specific prior written permission.
+#
+# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
+# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
+# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
+# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
+# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
+# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
+# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
+# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
+# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+#
+# Authors: Andreas Hansson
+
+try:
+
+ from mpl_toolkits.mplot3d import Axes3D
+ from matplotlib import cm
+ from matplotlib.ticker import LinearLocator, FormatStrFormatter
+ import matplotlib.pyplot as plt
+ import numpy as np
+except ImportError:
+ print "Failed to import matplotlib and numpy"
+ exit(-1)
+
+import sys
+import re
+
+# Determine the parameters of the sweep from the simout output, and
+# then parse the stats and plot the 3D surface corresponding to the
+# different combinations of parallel banks, and stride size, as
+# generated by the config/dram/sweep.py script
+def main():
+
+ if len(sys.argv) != 2:
+ print "Usage: ", sys.argv[0], " <simout directory>"
+ exit(-1)
+
+ try:
+ stats = open(sys.argv[1] + '/stats.txt', 'r')
+ except IOError:
+ print "Failed to open ", sys.argv[1] + '/stats.txt', " for reading"
+ exit(-1)
+
+ try:
+ simout = open(sys.argv[1] + '/simout', 'r')
+ except IOError:
+ print "Failed to open ", sys.argv[1] + '/simout', " for reading"
+ exit(-1)
+
+ # Get the burst size, number of banks and the maximum stride from
+ # the simulation output
+ got_sweep = False
+
+ for line in simout:
+ match = re.match("DRAM sweep with "
+ "burst: (\d+), banks: (\d+), max stride: (\d+)", line)
+ if match:
+ burst_size = int(match.groups(0)[0])
+ banks = int(match.groups(0)[1])
+ max_size = int(match.groups(0)[2])
+ got_sweep = True
+
+ simout.close()
+
+ if not got_sweep:
+ print "Failed to establish sweep details, ensure simout is up-to-date"
+ exit(-1)
+
+
+ # Collect the bus utilisation as our Z-axis, we do this in a 2D
+ # grid corresponding to each iteration over the various stride
+ # sizes.
+ z = []
+ zs = []
+ i = 0
+
+ # Now parse the stats
+ for line in stats:
+ match = re.match(".*busUtil\s+(\d+\.\d+)\s+#.*", line)
+ if match:
+ bus_util = float(match.groups(0)[0])
+ z.append(bus_util)
+ i += 1
+ # If we have completed a sweep over the stride sizes,
+ # start anew
+ if i == max_size / burst_size:
+ zs.append(z)
+ z = []
+ i = 0
+
+ stats.close()
+
+ # We should have a 2D grid with as many columns as banks
+ if len(zs) != banks:
+ print "Unexpected number of data points in stats output"
+ exit(-1)
+
+ fig = plt.figure()
+ ax = fig.gca(projection='3d')
+ X = np.arange(burst_size, max_size + 1, burst_size)
+ Y = np.arange(1, banks + 1, 1)
+ X, Y = np.meshgrid(X, Y)
+
+ # the values in the util are banks major, so we see groups for each
+ # stride size in order
+ Z = np.array(zs)
+
+ surf = ax.plot_surface(X, Y, Z, rstride=1, cstride=1, cmap=cm.coolwarm,
+ linewidth=0, antialiased=False)
+
+ # Change the tick frequency to 64
+ start, end = ax.get_xlim()
+ ax.xaxis.set_ticks(np.arange(start, end + 1, 64))
+
+ ax.set_xlabel('Bytes per activate')
+ ax.set_ylabel('Banks')
+ ax.set_zlabel('Efficiency (%)')
+
+ # Add a colorbar
+ fig.colorbar(surf, shrink=0.5, pad=.1, aspect=10)
+
+ plt.show()
+
+if __name__ == "__main__":
+ main()