Skip to content

GitLab

"scripts/git@developer.sourcefind.cn:zhaoyu6/sglang.git" did not exist on "712216928fa252d6592a1518579018a69cb72bfe"
Commit 5ca85c24 authored by Marvin Meiers's avatar Marvin Meiers Committed by Antoine Kaufmann
Browse files

experiments: add reference to network in E2ETopologyNode

parent c7300b4f
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
experiments/simbricks/orchestration/e2e_components.py
View file @ 5ca85c24
...@@ -159,6 +159,7 @@ class E2ETopologyNode(E2EComponent): ...@@ -159,6 +159,7 @@ class E2ETopologyNode(E2EComponent):
def __init__(self, idd: str) -> None: def __init__(self, idd: str) -> None:
super().__init__(idd) super().__init__(idd)
self.category = "TopologyNode" self.category = "TopologyNode"
self.network = None
class E2ESwitchNode(E2ETopologyNode): class E2ESwitchNode(E2ETopologyNode):
......
experiments/simbricks/orchestration/simulators.py
View file @ 5ca85c24
...@@ -909,6 +909,8 @@ class NS3E2ENet(NetSim): ...@@ -909,6 +909,8 @@ class NS3E2ENet(NetSim):
self.e2e_topologies.append(component) self.e2e_topologies.append(component)
else: else:
self.e2e_components.append(component) self.e2e_components.append(component)
if isinstance(component, e2e.E2ETopologyNode):
component.network = self
def resolve_socket_paths( def resolve_socket_paths(
self, self,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment