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 # int_links do not recycle buffers, so this parameter is not used. 45 # TODO: Move recycle_latency out of MessageBuffers and into controllers 46 recycle_latency = Param.Cycles(0, "") |
47
|
48 def setup_buffers(self): 49 # Note that all SimpleNetwork MessageBuffers are currently ordered 50 network_buffers = [] 51 for link in self.int_links: 52 # The network needs number_of_virtual_networks buffers per 53 # int_link port 54 for i in xrange(self.number_of_virtual_networks): 55 network_buffers.append(MessageBuffer(ordered = True)) 56 network_buffers.append(MessageBuffer(ordered = True)) 57 self.int_link_buffers = network_buffers 58 59 # Also add buffers for all router-link connections 60 for router in self.routers: 61 router_buffers = [] 62 # Add message buffers to routers for each internal link connection 63 for link in self.int_links: 64 if link.node_a == router: 65 for i in xrange(self.number_of_virtual_networks): 66 router_buffers.append(MessageBuffer(ordered = True)) 67 if link.node_b == router: 68 for i in xrange(self.number_of_virtual_networks): 69 router_buffers.append(MessageBuffer(ordered = True)) 70 71 # Add message buffers to routers for each external link connection 72 for link in self.ext_links: 73 # Routers can only be int_nodes on ext_links 74 if link.int_node in self.routers: 75 for i in xrange(self.number_of_virtual_networks): 76 router_buffers.append(MessageBuffer(ordered = True)) 77 router.port_buffers = router_buffers 78 |
79class Switch(BasicRouter): 80 type = 'Switch' 81 cxx_header = 'mem/ruby/network/simple/Switch.hh' 82 virt_nets = Param.Int(Parent.number_of_virtual_networks, 83 "number of virtual networks")
|
84 port_buffers = VectorParam.MessageBuffer("Port buffers") 85 # Ports do not recycle buffers, so this parameter is not used. 86 # TODO: Move recycle_latency out of MessageBuffers and into controllers 87 recycle_latency = Param.Cycles(0, "") |
|