SimpleNetwork.py (9389:8f8c911ab5a7) SimpleNetwork.py (11021:e8a6637afa4c)
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
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
34
35class SimpleNetwork(RubyNetwork):
36 type = 'SimpleNetwork'
37 cxx_header = "mem/ruby/network/simple/SimpleNetwork.hh"
38 buffer_size = Param.Int(0,
39 "default buffer size; 0 indicates infinite buffering");
40 endpoint_bandwidth = Param.Int(1000, "bandwidth adjustment factor");
41 adaptive_routing = Param.Bool(False, "enable adaptive routing");
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, "")
42
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
43class Switch(BasicRouter):
44 type = 'Switch'
45 cxx_header = 'mem/ruby/network/simple/Switch.hh'
46 virt_nets = Param.Int(Parent.number_of_virtual_networks,
47 "number of virtual networks")
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, "")