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 * 3213665Sandreas.sandberg@arm.com 3313665Sandreas.sandberg@arm.comfrom m5.objects.Network import RubyNetwork 3413665Sandreas.sandberg@arm.comfrom m5.objects.BasicRouter import BasicRouter 3513665Sandreas.sandberg@arm.comfrom m5.objects.MessageBuffer import MessageBuffer 366876Ssteve.reinhardt@amd.com 376876Ssteve.reinhardt@amd.comclass SimpleNetwork(RubyNetwork): 386876Ssteve.reinhardt@amd.com type = 'SimpleNetwork' 399338SAndreas.Sandberg@arm.com cxx_header = "mem/ruby/network/simple/SimpleNetwork.hh" 408259SBrad.Beckmann@amd.com buffer_size = Param.Int(0, 418259SBrad.Beckmann@amd.com "default buffer size; 0 indicates infinite buffering"); 428261SBrad.Beckmann@amd.com endpoint_bandwidth = Param.Int(1000, "bandwidth adjustment factor"); 438259SBrad.Beckmann@amd.com adaptive_routing = Param.Bool(False, "enable adaptive routing"); 4411021Sjthestness@gmail.com int_link_buffers = VectorParam.MessageBuffer("Buffers for int_links") 4511021Sjthestness@gmail.com 4611021Sjthestness@gmail.com def setup_buffers(self): 4711021Sjthestness@gmail.com # Note that all SimpleNetwork MessageBuffers are currently ordered 4811021Sjthestness@gmail.com network_buffers = [] 4911021Sjthestness@gmail.com for link in self.int_links: 5011021Sjthestness@gmail.com # The network needs number_of_virtual_networks buffers per 5111021Sjthestness@gmail.com # int_link port 5213709Sandreas.sandberg@arm.com for i in range(int(self.number_of_virtual_networks)): 5311021Sjthestness@gmail.com network_buffers.append(MessageBuffer(ordered = True)) 5411021Sjthestness@gmail.com network_buffers.append(MessageBuffer(ordered = True)) 5511021Sjthestness@gmail.com self.int_link_buffers = network_buffers 5611021Sjthestness@gmail.com 5711021Sjthestness@gmail.com # Also add buffers for all router-link connections 5811021Sjthestness@gmail.com for router in self.routers: 5911021Sjthestness@gmail.com router_buffers = [] 6011663Stushar@ece.gatech.edu # Add message buffers to routers at the end of each 6111663Stushar@ece.gatech.edu # unidirectional internal link 6211021Sjthestness@gmail.com for link in self.int_links: 6311663Stushar@ece.gatech.edu if link.dst_node == router: 6413709Sandreas.sandberg@arm.com for i in range(int(self.number_of_virtual_networks)): 6511021Sjthestness@gmail.com router_buffers.append(MessageBuffer(ordered = True)) 6611021Sjthestness@gmail.com 6711021Sjthestness@gmail.com # Add message buffers to routers for each external link connection 6811021Sjthestness@gmail.com for link in self.ext_links: 6911021Sjthestness@gmail.com # Routers can only be int_nodes on ext_links 7011021Sjthestness@gmail.com if link.int_node in self.routers: 7113709Sandreas.sandberg@arm.com for i in range(int(self.number_of_virtual_networks)): 7211021Sjthestness@gmail.com router_buffers.append(MessageBuffer(ordered = True)) 7311021Sjthestness@gmail.com router.port_buffers = router_buffers 749274Snilay@cs.wisc.edu 759274Snilay@cs.wisc.educlass Switch(BasicRouter): 769274Snilay@cs.wisc.edu type = 'Switch' 779389Sandreas.hansson@arm.com cxx_header = 'mem/ruby/network/simple/Switch.hh' 789274Snilay@cs.wisc.edu virt_nets = Param.Int(Parent.number_of_virtual_networks, 799274Snilay@cs.wisc.edu "number of virtual networks") 8011021Sjthestness@gmail.com port_buffers = VectorParam.MessageBuffer("Port buffers") 81