SimpleNetwork.py revision 11021
17019SBrad.Beckmann@amd.com# Copyright (c) 2009 Advanced Micro Devices, Inc. 27019SBrad.Beckmann@amd.com# All rights reserved. 37019SBrad.Beckmann@amd.com# 47019SBrad.Beckmann@amd.com# Redistribution and use in source and binary forms, with or without 57019SBrad.Beckmann@amd.com# modification, are permitted provided that the following conditions are 67019SBrad.Beckmann@amd.com# met: redistributions of source code must retain the above copyright 77019SBrad.Beckmann@amd.com# notice, this list of conditions and the following disclaimer; 87019SBrad.Beckmann@amd.com# redistributions in binary form must reproduce the above copyright 97019SBrad.Beckmann@amd.com# notice, this list of conditions and the following disclaimer in the 107019SBrad.Beckmann@amd.com# documentation and/or other materials provided with the distribution; 117019SBrad.Beckmann@amd.com# neither the name of the copyright holders nor the names of its 127019SBrad.Beckmann@amd.com# contributors may be used to endorse or promote products derived from 137019SBrad.Beckmann@amd.com# this software without specific prior written permission. 147019SBrad.Beckmann@amd.com# 157019SBrad.Beckmann@amd.com# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 167019SBrad.Beckmann@amd.com# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 177019SBrad.Beckmann@amd.com# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR 187019SBrad.Beckmann@amd.com# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT 197019SBrad.Beckmann@amd.com# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 207019SBrad.Beckmann@amd.com# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT 217019SBrad.Beckmann@amd.com# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 227019SBrad.Beckmann@amd.com# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 237019SBrad.Beckmann@amd.com# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 247019SBrad.Beckmann@amd.com# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 257019SBrad.Beckmann@amd.com# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 267019SBrad.Beckmann@amd.com# 277019SBrad.Beckmann@amd.com# Authors: Steve Reinhardt 287019SBrad.Beckmann@amd.com# Brad Beckmann 297019SBrad.Beckmann@amd.com 306876Ssteve.reinhardt@amd.comfrom m5.params import * 319274Snilay@cs.wisc.edufrom m5.proxy import * 326876Ssteve.reinhardt@amd.comfrom Network import RubyNetwork 339274Snilay@cs.wisc.edufrom BasicRouter import BasicRouter 3411021Sjthestness@gmail.comfrom MessageBuffer import MessageBuffer 356876Ssteve.reinhardt@amd.com 366876Ssteve.reinhardt@amd.comclass SimpleNetwork(RubyNetwork): 376876Ssteve.reinhardt@amd.com type = 'SimpleNetwork' 389338SAndreas.Sandberg@arm.com cxx_header = "mem/ruby/network/simple/SimpleNetwork.hh" 398259SBrad.Beckmann@amd.com buffer_size = Param.Int(0, 408259SBrad.Beckmann@amd.com "default buffer size; 0 indicates infinite buffering"); 418261SBrad.Beckmann@amd.com endpoint_bandwidth = Param.Int(1000, "bandwidth adjustment factor"); 428259SBrad.Beckmann@amd.com adaptive_routing = Param.Bool(False, "enable adaptive routing"); 4311021Sjthestness@gmail.com int_link_buffers = VectorParam.MessageBuffer("Buffers for int_links") 4411021Sjthestness@gmail.com # int_links do not recycle buffers, so this parameter is not used. 4511021Sjthestness@gmail.com # TODO: Move recycle_latency out of MessageBuffers and into controllers 4611021Sjthestness@gmail.com recycle_latency = Param.Cycles(0, "") 4711021Sjthestness@gmail.com 4811021Sjthestness@gmail.com def setup_buffers(self): 4911021Sjthestness@gmail.com # Note that all SimpleNetwork MessageBuffers are currently ordered 5011021Sjthestness@gmail.com network_buffers = [] 5111021Sjthestness@gmail.com for link in self.int_links: 5211021Sjthestness@gmail.com # The network needs number_of_virtual_networks buffers per 5311021Sjthestness@gmail.com # int_link port 5411021Sjthestness@gmail.com for i in xrange(self.number_of_virtual_networks): 5511021Sjthestness@gmail.com network_buffers.append(MessageBuffer(ordered = True)) 5611021Sjthestness@gmail.com network_buffers.append(MessageBuffer(ordered = True)) 5711021Sjthestness@gmail.com self.int_link_buffers = network_buffers 5811021Sjthestness@gmail.com 5911021Sjthestness@gmail.com # Also add buffers for all router-link connections 6011021Sjthestness@gmail.com for router in self.routers: 6111021Sjthestness@gmail.com router_buffers = [] 6211021Sjthestness@gmail.com # Add message buffers to routers for each internal link connection 6311021Sjthestness@gmail.com for link in self.int_links: 6411021Sjthestness@gmail.com if link.node_a == router: 6511021Sjthestness@gmail.com for i in xrange(self.number_of_virtual_networks): 6611021Sjthestness@gmail.com router_buffers.append(MessageBuffer(ordered = True)) 6711021Sjthestness@gmail.com if link.node_b == router: 6811021Sjthestness@gmail.com for i in xrange(self.number_of_virtual_networks): 6911021Sjthestness@gmail.com router_buffers.append(MessageBuffer(ordered = True)) 7011021Sjthestness@gmail.com 7111021Sjthestness@gmail.com # Add message buffers to routers for each external link connection 7211021Sjthestness@gmail.com for link in self.ext_links: 7311021Sjthestness@gmail.com # Routers can only be int_nodes on ext_links 7411021Sjthestness@gmail.com if link.int_node in self.routers: 7511021Sjthestness@gmail.com for i in xrange(self.number_of_virtual_networks): 7611021Sjthestness@gmail.com router_buffers.append(MessageBuffer(ordered = True)) 7711021Sjthestness@gmail.com router.port_buffers = router_buffers 789274Snilay@cs.wisc.edu 799274Snilay@cs.wisc.educlass Switch(BasicRouter): 809274Snilay@cs.wisc.edu type = 'Switch' 819389Sandreas.hansson@arm.com cxx_header = 'mem/ruby/network/simple/Switch.hh' 829274Snilay@cs.wisc.edu virt_nets = Param.Int(Parent.number_of_virtual_networks, 839274Snilay@cs.wisc.edu "number of virtual networks") 8411021Sjthestness@gmail.com port_buffers = VectorParam.MessageBuffer("Port buffers") 8511021Sjthestness@gmail.com # Ports do not recycle buffers, so this parameter is not used. 8611021Sjthestness@gmail.com # TODO: Move recycle_latency out of MessageBuffers and into controllers 8711021Sjthestness@gmail.com recycle_latency = Param.Cycles(0, "") 88