| # 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() |
