Commit 165fa8ea authored by Hejing Li's avatar Hejing Li
Browse files

add system and simulator mappings to experiment

parent 8f254b47
...@@ -24,6 +24,8 @@ import itertools ...@@ -24,6 +24,8 @@ import itertools
import typing as tp import typing as tp
from simbricks.orchestration import simulators from simbricks.orchestration import simulators
import simbricks.orchestration.simulation as simulation
import simbricks.orchestration.system as system
from simbricks.orchestration.proxy import NetProxyConnecter, NetProxyListener from simbricks.orchestration.proxy import NetProxyConnecter, NetProxyListener
from simbricks.orchestration.simulators import ( from simbricks.orchestration.simulators import (
HostSim, I40eMultiNIC, NetSim, NICSim, PCIDevSim, Simulator HostSim, I40eMultiNIC, NetSim, NICSim, PCIDevSim, Simulator
...@@ -69,6 +71,9 @@ class Experiment(object): ...@@ -69,6 +71,9 @@ class Experiment(object):
"""The network simulators to run.""" """The network simulators to run."""
self.metadata: tp.Dict[str, tp.Any] = {} self.metadata: tp.Dict[str, tp.Any] = {}
self.sys_sim_map: tp.Dict[system.Component, simulation.Simulator] = {}
"""System component and its simulator pairs"""
@property @property
def nics(self): def nics(self):
return filter(lambda pcidev: pcidev.is_nic(), self.pcidevs) return filter(lambda pcidev: pcidev.is_nic(), self.pcidevs)
......
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