MCPcopy Create free account
hub / github.com/FastMAS/KVCOMM / generate_layered_graph

Function generate_layered_graph

experiments/benchmark_TTFT.py:197–211  ·  view source on GitHub ↗
(n, layer_num=2)

Source from the content-addressed store, hash-verified

195 node_kwargs = None
196
197 def generate_layered_graph(n, layer_num=2):
198 adj_matrix = [[0] * n for _ in range(n)]
199 base_size = n // layer_num
200 remainder = n % layer_num
201 layers: List[int] = []
202 for i in range(layer_num):
203 size = base_size + (1 if i < remainder else 0)
204 layers.extend([i] * size)
205 random.shuffle(layers)
206 for i in range(n):
207 current_layer = layers[i]
208 for j in range(n):
209 if layers[j] == current_layer + 1:
210 adj_matrix[i][j] = 1
211 return adj_matrix
212
213 def generate_mesh_graph(n):
214 adj_matrix = [[0] * n for _ in range(n)]

Callers 1

get_kwargsFunction · 0.70

Calls

no outgoing calls

Tested by

no test coverage detected