1# Copyright (c) 2010 Advanced Micro Devices, Inc. 2# 2016 Georgia Institute of Technology 3# All rights reserved. 4# 5# Redistribution and use in source and binary forms, with or without 6# modification, are permitted provided that the following conditions are 7# met: redistributions of source code must retain the above copyright 8# notice, this list of conditions and the following disclaimer; --- 64 unchanged lines hidden (view full) --- 73 74 # link counter to set unique link ids 75 link_count = 0 76 77 # Add all but the remainder nodes to the list of nodes to be uniformly 78 # distributed across the network. 79 network_nodes = [] 80 remainder_nodes = [] |
81 for node_index in range(len(nodes)): |
82 if node_index < (len(nodes) - remainder): 83 network_nodes.append(nodes[node_index]) 84 else: 85 remainder_nodes.append(nodes[node_index]) 86 87 # Connect each node to the appropriate router 88 ext_links = [] 89 for (i, n) in enumerate(network_nodes): --- 15 unchanged lines hidden (view full) --- 105 link_count += 1 106 107 network.ext_links = ext_links 108 109 # Create the mesh links. 110 int_links = [] 111 112 # East output to West input links (weight = 1) |
113 for row in range(num_rows): 114 for col in range(num_columns): |
115 if (col + 1 < num_columns): 116 east_out = col + (row * num_columns) 117 west_in = (col + 1) + (row * num_columns) 118 int_links.append(IntLink(link_id=link_count, 119 src_node=routers[east_out], 120 dst_node=routers[west_in], 121 src_outport="East", 122 dst_inport="West", 123 latency = link_latency, 124 weight=1)) 125 link_count += 1 126 127 # West output to East input links (weight = 1) |
128 for row in range(num_rows): 129 for col in range(num_columns): |
130 if (col + 1 < num_columns): 131 east_in = col + (row * num_columns) 132 west_out = (col + 1) + (row * num_columns) 133 int_links.append(IntLink(link_id=link_count, 134 src_node=routers[west_out], 135 dst_node=routers[east_in], 136 src_outport="West", 137 dst_inport="East", 138 latency = link_latency, 139 weight=1)) 140 link_count += 1 141 142 # North output to South input links (weight = 2) |
143 for col in range(num_columns): 144 for row in range(num_rows): |
145 if (row + 1 < num_rows): 146 north_out = col + (row * num_columns) 147 south_in = col + ((row + 1) * num_columns) 148 int_links.append(IntLink(link_id=link_count, 149 src_node=routers[north_out], 150 dst_node=routers[south_in], 151 src_outport="North", 152 dst_inport="South", 153 latency = link_latency, 154 weight=2)) 155 link_count += 1 156 157 # South output to North input links (weight = 2) |
158 for col in range(num_columns): 159 for row in range(num_rows): |
160 if (row + 1 < num_rows): 161 north_in = col + (row * num_columns) 162 south_out = col + ((row + 1) * num_columns) 163 int_links.append(IntLink(link_id=link_count, 164 src_node=routers[south_out], 165 dst_node=routers[north_in], 166 src_outport="South", 167 dst_inport="North", 168 latency = link_latency, 169 weight=2)) 170 link_count += 1 171 172 173 network.int_links = int_links |