110449Snilay@cs.wisc.edu# Copyright (c) 2014 Mark D. Hill and David A. Wood 210449Snilay@cs.wisc.edu# All rights reserved. 310449Snilay@cs.wisc.edu# 410449Snilay@cs.wisc.edu# Redistribution and use in source and binary forms, with or without 510449Snilay@cs.wisc.edu# modification, are permitted provided that the following conditions are 610449Snilay@cs.wisc.edu# met: redistributions of source code must retain the above copyright 710449Snilay@cs.wisc.edu# notice, this list of conditions and the following disclaimer; 810449Snilay@cs.wisc.edu# redistributions in binary form must reproduce the above copyright 910449Snilay@cs.wisc.edu# notice, this list of conditions and the following disclaimer in the 1010449Snilay@cs.wisc.edu# documentation and/or other materials provided with the distribution; 1110449Snilay@cs.wisc.edu# neither the name of the copyright holders nor the names of its 1210449Snilay@cs.wisc.edu# contributors may be used to endorse or promote products derived from 1310449Snilay@cs.wisc.edu# this software without specific prior written permission. 1410449Snilay@cs.wisc.edu# 1510449Snilay@cs.wisc.edu# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 1610449Snilay@cs.wisc.edu# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 1710449Snilay@cs.wisc.edu# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR 1810449Snilay@cs.wisc.edu# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT 1910449Snilay@cs.wisc.edu# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 2010449Snilay@cs.wisc.edu# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT 2110449Snilay@cs.wisc.edu# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 2210449Snilay@cs.wisc.edu# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 2310449Snilay@cs.wisc.edu# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 2410449Snilay@cs.wisc.edu# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 2510449Snilay@cs.wisc.edu# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 2610449Snilay@cs.wisc.edu 2710449Snilay@cs.wisc.edu 2810449Snilay@cs.wisc.edufrom ConfigParser import ConfigParser 2910449Snilay@cs.wisc.eduimport string, sys, subprocess, os 3010449Snilay@cs.wisc.edu 3110449Snilay@cs.wisc.edu# Compile DSENT to generate the Python module and then import it. 3210449Snilay@cs.wisc.edu# This script assumes it is executed from the gem5 root. 3310449Snilay@cs.wisc.eduprint("Attempting compilation") 3410449Snilay@cs.wisc.edufrom subprocess import call 3510449Snilay@cs.wisc.edu 3610449Snilay@cs.wisc.edusrc_dir = 'ext/dsent' 3710449Snilay@cs.wisc.edubuild_dir = 'build/ext/dsent' 3810449Snilay@cs.wisc.edu 3910449Snilay@cs.wisc.eduif not os.path.exists(build_dir): 4010449Snilay@cs.wisc.edu os.makedirs(build_dir) 4110449Snilay@cs.wisc.eduos.chdir(build_dir) 4210449Snilay@cs.wisc.edu 4310449Snilay@cs.wisc.eduerror = call(['cmake', '../../../%s' % src_dir]) 4410449Snilay@cs.wisc.eduif error: 4510449Snilay@cs.wisc.edu print("Failed to run cmake") 4610449Snilay@cs.wisc.edu exit(-1) 4710449Snilay@cs.wisc.edu 4810449Snilay@cs.wisc.eduerror = call(['make']) 4910449Snilay@cs.wisc.eduif error: 5010449Snilay@cs.wisc.edu print("Failed to run make") 5110449Snilay@cs.wisc.edu exit(-1) 5210449Snilay@cs.wisc.edu 5310449Snilay@cs.wisc.eduprint("Compiled dsent") 5410449Snilay@cs.wisc.eduos.chdir("../../../") 5510449Snilay@cs.wisc.edusys.path.append("build/ext/dsent") 5610449Snilay@cs.wisc.eduimport dsent 5710449Snilay@cs.wisc.edu 5810449Snilay@cs.wisc.edu# Parse gem5 config.ini file for the configuration parameters related to 5910449Snilay@cs.wisc.edu# the on-chip network. 6010449Snilay@cs.wisc.edudef parseConfig(config_file): 6110449Snilay@cs.wisc.edu config = ConfigParser() 6210449Snilay@cs.wisc.edu if not config.read(config_file): 6310449Snilay@cs.wisc.edu print("ERROR: config file '", config_file, "' not found") 6410449Snilay@cs.wisc.edu sys.exit(1) 6510449Snilay@cs.wisc.edu 6610449Snilay@cs.wisc.edu if not config.has_section("system.ruby.network"): 6710449Snilay@cs.wisc.edu print("ERROR: Ruby network not found in '", config_file) 6810449Snilay@cs.wisc.edu sys.exit(1) 6910449Snilay@cs.wisc.edu 7010449Snilay@cs.wisc.edu if config.get("system.ruby.network", "type") != "GarnetNetwork_d" : 7110449Snilay@cs.wisc.edu print("ERROR: Garnet network not used in '", config_file) 7210449Snilay@cs.wisc.edu sys.exit(1) 7310449Snilay@cs.wisc.edu 7410449Snilay@cs.wisc.edu number_of_virtual_networks = config.getint("system.ruby.network", 7510449Snilay@cs.wisc.edu "number_of_virtual_networks") 7610449Snilay@cs.wisc.edu vcs_per_vnet = config.getint("system.ruby.network", "vcs_per_vnet") 7710449Snilay@cs.wisc.edu 7810449Snilay@cs.wisc.edu buffers_per_data_vc = config.getint("system.ruby.network", 7910449Snilay@cs.wisc.edu "buffers_per_data_vc") 8010449Snilay@cs.wisc.edu buffers_per_control_vc = config.getint("system.ruby.network", 8110449Snilay@cs.wisc.edu "buffers_per_ctrl_vc") 8210449Snilay@cs.wisc.edu 8310449Snilay@cs.wisc.edu ni_flit_size_bits = 8 * config.getint("system.ruby.network", 8410449Snilay@cs.wisc.edu "ni_flit_size") 8510449Snilay@cs.wisc.edu 8610449Snilay@cs.wisc.edu routers = config.get("system.ruby.network", "routers").split() 8710449Snilay@cs.wisc.edu int_links = config.get("system.ruby.network", "int_links").split() 8810449Snilay@cs.wisc.edu ext_links = config.get("system.ruby.network", "ext_links").split() 8910449Snilay@cs.wisc.edu 9010449Snilay@cs.wisc.edu return (config, number_of_virtual_networks, vcs_per_vnet, 9110449Snilay@cs.wisc.edu buffers_per_data_vc, buffers_per_control_vc, ni_flit_size_bits, 9210449Snilay@cs.wisc.edu routers, int_links, ext_links) 9310449Snilay@cs.wisc.edu 9410449Snilay@cs.wisc.edu 9510449Snilay@cs.wisc.edudef getClock(obj, config): 9610449Snilay@cs.wisc.edu if config.get(obj, "type") == "SrcClockDomain": 9710449Snilay@cs.wisc.edu return config.getint(obj, "clock") 9810449Snilay@cs.wisc.edu 9910449Snilay@cs.wisc.edu if config.get(obj, "type") == "DerivedClockDomain": 10010449Snilay@cs.wisc.edu source = config.get(obj, "clk_domain") 10110449Snilay@cs.wisc.edu divider = config.getint(obj, "clk_divider") 10210449Snilay@cs.wisc.edu return getClock(source, config) / divider 10310449Snilay@cs.wisc.edu 10410449Snilay@cs.wisc.edu source = config.get(obj, "clk_domain") 10510449Snilay@cs.wisc.edu return getClock(source, config) 10610449Snilay@cs.wisc.edu 10710449Snilay@cs.wisc.edu 10810449Snilay@cs.wisc.edu## Compute the power consumed by the given router 10910449Snilay@cs.wisc.edudef computeRouterPowerAndArea(router, stats_file, config, int_links, ext_links, 11010449Snilay@cs.wisc.edu number_of_virtual_networks, vcs_per_vnet, 11110449Snilay@cs.wisc.edu buffers_per_data_vc, buffers_per_control_vc, 11210449Snilay@cs.wisc.edu ni_flit_size_bits): 11310449Snilay@cs.wisc.edu frequency = getClock(router, config) 11410449Snilay@cs.wisc.edu num_ports = 0 11510449Snilay@cs.wisc.edu 11610449Snilay@cs.wisc.edu for int_link in int_links: 11710449Snilay@cs.wisc.edu if config.get(int_link, "node_a") == router or \ 11810449Snilay@cs.wisc.edu config.get(int_link, "node_b") == router: 11910449Snilay@cs.wisc.edu num_ports += 1 12010449Snilay@cs.wisc.edu 12110449Snilay@cs.wisc.edu for ext_link in ext_links: 12210449Snilay@cs.wisc.edu if config.get(ext_link, "int_node") == router: 12310449Snilay@cs.wisc.edu num_ports += 1 12410449Snilay@cs.wisc.edu 12510449Snilay@cs.wisc.edu power = dsent.computeRouterPowerAndArea(frequency, num_ports, num_ports, 12610449Snilay@cs.wisc.edu number_of_virtual_networks, 12710449Snilay@cs.wisc.edu vcs_per_vnet, buffers_per_data_vc, 12810449Snilay@cs.wisc.edu ni_flit_size_bits) 12910449Snilay@cs.wisc.edu 13010449Snilay@cs.wisc.edu print("%s Power: " % router, power) 13110449Snilay@cs.wisc.edu 13210449Snilay@cs.wisc.edu 13310449Snilay@cs.wisc.edu## Compute the power consumed by the given link 13410449Snilay@cs.wisc.edudef computeLinkPower(link, stats_file, config, sim_seconds): 13510449Snilay@cs.wisc.edu frequency = getClock(link + ".nls0", config) 13610449Snilay@cs.wisc.edu power = dsent.computeLinkPower(frequency) 13710449Snilay@cs.wisc.edu print("%s.nls0 Power: " % link, power) 13810449Snilay@cs.wisc.edu 13910449Snilay@cs.wisc.edu frequency = getClock(link + ".nls1", config) 14010449Snilay@cs.wisc.edu power = dsent.computeLinkPower(frequency) 14110449Snilay@cs.wisc.edu print("%s.nls1 Power: " % link, power) 14210449Snilay@cs.wisc.edu 14310449Snilay@cs.wisc.edu 14410449Snilay@cs.wisc.edudef parseStats(stats_file, config, router_config_file, link_config_file, 14510449Snilay@cs.wisc.edu routers, int_links, ext_links, number_of_virtual_networks, 14610449Snilay@cs.wisc.edu vcs_per_vnet, buffers_per_data_vc, buffers_per_control_vc, 14710449Snilay@cs.wisc.edu ni_flit_size_bits): 14810449Snilay@cs.wisc.edu 14910449Snilay@cs.wisc.edu # Open the stats.txt file and parse it to for the required numbers 15010449Snilay@cs.wisc.edu # and the number of routers. 15110449Snilay@cs.wisc.edu try: 15210449Snilay@cs.wisc.edu stats_handle = open(stats_file, 'r') 15310449Snilay@cs.wisc.edu stats_handle.close() 15410449Snilay@cs.wisc.edu except IOError: 15510449Snilay@cs.wisc.edu print("Failed to open ", stats_file, " for reading") 15610449Snilay@cs.wisc.edu exit(-1) 15710449Snilay@cs.wisc.edu 15810449Snilay@cs.wisc.edu # Now parse the stats 15910449Snilay@cs.wisc.edu pattern = "sim_seconds" 16010449Snilay@cs.wisc.edu lines = string.split(subprocess.check_output( 16110449Snilay@cs.wisc.edu ["grep", pattern, stats_file]), '\n', -1) 16210449Snilay@cs.wisc.edu assert len(lines) >= 1 16310449Snilay@cs.wisc.edu 16410449Snilay@cs.wisc.edu ## Assume that the first line is the one required 16510449Snilay@cs.wisc.edu [l1,l2,l3] = lines[0].partition(" ") 16610449Snilay@cs.wisc.edu l4 = l3.strip().partition(" ") 16710449Snilay@cs.wisc.edu simulation_length_in_seconds = float(l4[0]) 16810449Snilay@cs.wisc.edu 16910449Snilay@cs.wisc.edu # Initialize DSENT with a configuration file 17010449Snilay@cs.wisc.edu dsent.initialize(router_config_file) 17110449Snilay@cs.wisc.edu 17210449Snilay@cs.wisc.edu # Compute the power consumed by the routers 17310449Snilay@cs.wisc.edu for router in routers: 17410449Snilay@cs.wisc.edu computeRouterPowerAndArea(router, stats_file, config, int_links, 17510449Snilay@cs.wisc.edu ext_links, number_of_virtual_networks, 17610449Snilay@cs.wisc.edu vcs_per_vnet, buffers_per_data_vc, 17710449Snilay@cs.wisc.edu buffers_per_control_vc, ni_flit_size_bits) 17810449Snilay@cs.wisc.edu 17910449Snilay@cs.wisc.edu # Finalize DSENT 18010449Snilay@cs.wisc.edu dsent.finalize() 18110449Snilay@cs.wisc.edu 18210449Snilay@cs.wisc.edu # Initialize DSENT with a configuration file 18310449Snilay@cs.wisc.edu dsent.initialize(link_config_file) 18410449Snilay@cs.wisc.edu 18510449Snilay@cs.wisc.edu # Compute the power consumed by the links 18610449Snilay@cs.wisc.edu for link in int_links: 18710449Snilay@cs.wisc.edu computeLinkPower(link, stats_file, config, 18810449Snilay@cs.wisc.edu simulation_length_in_seconds) 18910449Snilay@cs.wisc.edu for link in ext_links: 19010449Snilay@cs.wisc.edu computeLinkPower(link, stats_file, config, 19110449Snilay@cs.wisc.edu simulation_length_in_seconds) 19210449Snilay@cs.wisc.edu 19310449Snilay@cs.wisc.edu # Finalize DSENT 19410449Snilay@cs.wisc.edu dsent.finalize() 19510449Snilay@cs.wisc.edu 19610449Snilay@cs.wisc.edu# This script parses the config.ini and the stats.txt from a run and 19710449Snilay@cs.wisc.edu# generates the power and the area of the on-chip network using DSENT 19810449Snilay@cs.wisc.edudef main(): 19910449Snilay@cs.wisc.edu if len(sys.argv) != 5: 20010449Snilay@cs.wisc.edu print("Usage: ", sys.argv[0], " <gem5 root directory> " \ 20110449Snilay@cs.wisc.edu "<simulation directory> <router config file> <link config file>") 20210449Snilay@cs.wisc.edu exit(-1) 20310449Snilay@cs.wisc.edu 20410449Snilay@cs.wisc.edu print("WARNING: configuration files for DSENT and McPAT are separate. " \ 20510449Snilay@cs.wisc.edu "Changes made to one are not reflected in the other.") 20610449Snilay@cs.wisc.edu 20710449Snilay@cs.wisc.edu (config, number_of_virtual_networks, vcs_per_vnet, buffers_per_data_vc, 20810449Snilay@cs.wisc.edu buffers_per_control_vc, ni_flit_size_bits, routers, int_links, 20910449Snilay@cs.wisc.edu ext_links) = parseConfig("%s/%s/config.ini" % (sys.argv[1], sys.argv[2])) 21010449Snilay@cs.wisc.edu 21110449Snilay@cs.wisc.edu parseStats("%s/%s/stats.txt" % (sys.argv[1], sys.argv[2]), config, 21210449Snilay@cs.wisc.edu sys.argv[3], sys.argv[4], routers, int_links, ext_links, 21310449Snilay@cs.wisc.edu number_of_virtual_networks, vcs_per_vnet, buffers_per_data_vc, 21410449Snilay@cs.wisc.edu buffers_per_control_vc, ni_flit_size_bits) 21510449Snilay@cs.wisc.edu 21610449Snilay@cs.wisc.eduif __name__ == "__main__": 21710449Snilay@cs.wisc.edu main() 218