SimpleNetwork.py revision 11663
1# Copyright (c) 2009 Advanced Micro Devices, Inc. 2# All rights reserved. 3# 4# Redistribution and use in source and binary forms, with or without 5# modification, are permitted provided that the following conditions are 6# met: redistributions of source code must retain the above copyright 7# notice, this list of conditions and the following disclaimer; 8# redistributions in binary form must reproduce the above copyright 9# notice, this list of conditions and the following disclaimer in the 10# documentation and/or other materials provided with the distribution; 11# neither the name of the copyright holders nor the names of its 12# contributors may be used to endorse or promote products derived from 13# this software without specific prior written permission. 14# 15# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 16# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 17# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR 18# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT 19# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 20# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT 21# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 22# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 23# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 24# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 25# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 26# 27# Authors: Steve Reinhardt 28# Brad Beckmann 29 30from m5.params import * 31from m5.proxy import * 32from Network import RubyNetwork 33from BasicRouter import BasicRouter 34from MessageBuffer import MessageBuffer 35 36class SimpleNetwork(RubyNetwork): 37 type = 'SimpleNetwork' 38 cxx_header = "mem/ruby/network/simple/SimpleNetwork.hh" 39 buffer_size = Param.Int(0, 40 "default buffer size; 0 indicates infinite buffering"); 41 endpoint_bandwidth = Param.Int(1000, "bandwidth adjustment factor"); 42 adaptive_routing = Param.Bool(False, "enable adaptive routing"); 43 int_link_buffers = VectorParam.MessageBuffer("Buffers for int_links") 44 45 def setup_buffers(self): 46 # Note that all SimpleNetwork MessageBuffers are currently ordered 47 network_buffers = [] 48 for link in self.int_links: 49 # The network needs number_of_virtual_networks buffers per 50 # int_link port 51 for i in xrange(self.number_of_virtual_networks): 52 network_buffers.append(MessageBuffer(ordered = True)) 53 network_buffers.append(MessageBuffer(ordered = True)) 54 self.int_link_buffers = network_buffers 55 56 # Also add buffers for all router-link connections 57 for router in self.routers: 58 router_buffers = [] 59 # Add message buffers to routers at the end of each 60 # unidirectional internal link 61 for link in self.int_links: 62 if link.dst_node == router: 63 for i in xrange(self.number_of_virtual_networks): 64 router_buffers.append(MessageBuffer(ordered = True)) 65 66 # Add message buffers to routers for each external link connection 67 for link in self.ext_links: 68 # Routers can only be int_nodes on ext_links 69 if link.int_node in self.routers: 70 for i in xrange(self.number_of_virtual_networks): 71 router_buffers.append(MessageBuffer(ordered = True)) 72 router.port_buffers = router_buffers 73 74class Switch(BasicRouter): 75 type = 'Switch' 76 cxx_header = 'mem/ruby/network/simple/Switch.hh' 77 virt_nets = Param.Int(Parent.number_of_virtual_networks, 78 "number of virtual networks") 79 port_buffers = VectorParam.MessageBuffer("Port buffers") 80