util: adds a script for using DSENT

This patch adds a python script that processes the configuration and the
statistics file from a simulation run.  Configuration and activity of network
routers and links obtained from this processing is fed to DSENT via its Python
interface.  DSENT then computes the area and the power consumption of these
network components.  The script outputs these quantities to the console.
This commit is contained in:
Nilay Vaish 2014-10-11 16:16:00 -05:00
parent c1aecc05e6
commit 19ca35737d

View file

@ -0,0 +1,217 @@
# Copyright (c) 2014 Mark D. Hill and David A. Wood
# All rights reserved.
#
# 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.
from ConfigParser import ConfigParser
import string, sys, subprocess, os
# Compile DSENT to generate the Python module and then import it.
# This script assumes it is executed from the gem5 root.
print("Attempting compilation")
from subprocess import call
src_dir = 'ext/dsent'
build_dir = 'build/ext/dsent'
if not os.path.exists(build_dir):
os.makedirs(build_dir)
os.chdir(build_dir)
error = call(['cmake', '../../../%s' % src_dir])
if error:
print("Failed to run cmake")
exit(-1)
error = call(['make'])
if error:
print("Failed to run make")
exit(-1)
print("Compiled dsent")
os.chdir("../../../")
sys.path.append("build/ext/dsent")
import dsent
# Parse gem5 config.ini file for the configuration parameters related to
# the on-chip network.
def parseConfig(config_file):
config = ConfigParser()
if not config.read(config_file):
print("ERROR: config file '", config_file, "' not found")
sys.exit(1)
if not config.has_section("system.ruby.network"):
print("ERROR: Ruby network not found in '", config_file)
sys.exit(1)
if config.get("system.ruby.network", "type") != "GarnetNetwork_d" :
print("ERROR: Garnet network not used in '", config_file)
sys.exit(1)
number_of_virtual_networks = config.getint("system.ruby.network",
"number_of_virtual_networks")
vcs_per_vnet = config.getint("system.ruby.network", "vcs_per_vnet")
buffers_per_data_vc = config.getint("system.ruby.network",
"buffers_per_data_vc")
buffers_per_control_vc = config.getint("system.ruby.network",
"buffers_per_ctrl_vc")
ni_flit_size_bits = 8 * config.getint("system.ruby.network",
"ni_flit_size")
routers = config.get("system.ruby.network", "routers").split()
int_links = config.get("system.ruby.network", "int_links").split()
ext_links = config.get("system.ruby.network", "ext_links").split()
return (config, number_of_virtual_networks, vcs_per_vnet,
buffers_per_data_vc, buffers_per_control_vc, ni_flit_size_bits,
routers, int_links, ext_links)
def getClock(obj, config):
if config.get(obj, "type") == "SrcClockDomain":
return config.getint(obj, "clock")
if config.get(obj, "type") == "DerivedClockDomain":
source = config.get(obj, "clk_domain")
divider = config.getint(obj, "clk_divider")
return getClock(source, config) / divider
source = config.get(obj, "clk_domain")
return getClock(source, config)
## Compute the power consumed by the given router
def computeRouterPowerAndArea(router, stats_file, config, int_links, ext_links,
number_of_virtual_networks, vcs_per_vnet,
buffers_per_data_vc, buffers_per_control_vc,
ni_flit_size_bits):
frequency = getClock(router, config)
num_ports = 0
for int_link in int_links:
if config.get(int_link, "node_a") == router or \
config.get(int_link, "node_b") == router:
num_ports += 1
for ext_link in ext_links:
if config.get(ext_link, "int_node") == router:
num_ports += 1
power = dsent.computeRouterPowerAndArea(frequency, num_ports, num_ports,
number_of_virtual_networks,
vcs_per_vnet, buffers_per_data_vc,
ni_flit_size_bits)
print("%s Power: " % router, power)
## Compute the power consumed by the given link
def computeLinkPower(link, stats_file, config, sim_seconds):
frequency = getClock(link + ".nls0", config)
power = dsent.computeLinkPower(frequency)
print("%s.nls0 Power: " % link, power)
frequency = getClock(link + ".nls1", config)
power = dsent.computeLinkPower(frequency)
print("%s.nls1 Power: " % link, power)
def parseStats(stats_file, config, router_config_file, link_config_file,
routers, int_links, ext_links, number_of_virtual_networks,
vcs_per_vnet, buffers_per_data_vc, buffers_per_control_vc,
ni_flit_size_bits):
# Open the stats.txt file and parse it to for the required numbers
# and the number of routers.
try:
stats_handle = open(stats_file, 'r')
stats_handle.close()
except IOError:
print("Failed to open ", stats_file, " for reading")
exit(-1)
# Now parse the stats
pattern = "sim_seconds"
lines = string.split(subprocess.check_output(
["grep", pattern, stats_file]), '\n', -1)
assert len(lines) >= 1
## Assume that the first line is the one required
[l1,l2,l3] = lines[0].partition(" ")
l4 = l3.strip().partition(" ")
simulation_length_in_seconds = float(l4[0])
# Initialize DSENT with a configuration file
dsent.initialize(router_config_file)
# Compute the power consumed by the routers
for router in routers:
computeRouterPowerAndArea(router, stats_file, config, int_links,
ext_links, number_of_virtual_networks,
vcs_per_vnet, buffers_per_data_vc,
buffers_per_control_vc, ni_flit_size_bits)
# Finalize DSENT
dsent.finalize()
# Initialize DSENT with a configuration file
dsent.initialize(link_config_file)
# Compute the power consumed by the links
for link in int_links:
computeLinkPower(link, stats_file, config,
simulation_length_in_seconds)
for link in ext_links:
computeLinkPower(link, stats_file, config,
simulation_length_in_seconds)
# Finalize DSENT
dsent.finalize()
# This script parses the config.ini and the stats.txt from a run and
# generates the power and the area of the on-chip network using DSENT
def main():
if len(sys.argv) != 5:
print("Usage: ", sys.argv[0], " <gem5 root directory> " \
"<simulation directory> <router config file> <link config file>")
exit(-1)
print("WARNING: configuration files for DSENT and McPAT are separate. " \
"Changes made to one are not reflected in the other.")
(config, number_of_virtual_networks, vcs_per_vnet, buffers_per_data_vc,
buffers_per_control_vc, ni_flit_size_bits, routers, int_links,
ext_links) = parseConfig("%s/%s/config.ini" % (sys.argv[1], sys.argv[2]))
parseStats("%s/%s/stats.txt" % (sys.argv[1], sys.argv[2]), config,
sys.argv[3], sys.argv[4], routers, int_links, ext_links,
number_of_virtual_networks, vcs_per_vnet, buffers_per_data_vc,
buffers_per_control_vc, ni_flit_size_bits)
if __name__ == "__main__":
main()