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 --- 17 unchanged lines hidden (view full) --- 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, "") |