fix auto_dist missing nic attribute on Experiment and HostSim

experiments/simbricks/experiments.py

@@ -59,6 +59,10 @@ class Experiment(object):
         """The network simulators to run."""
         self.metadata = {}
 
+    @property
+    def nics(self):
+        return filter(lambda pcidev: pcidev.is_nic(), self.pcidevs)
+
     def add_host(self, sim: HostSim):
         for h in self.hosts:
             if h.name == sim.name:

experiments/simbricks/runtime/distributed.py

@@ -140,14 +140,14 @@ def auto_dist(
     for h in e.hosts:
         de.add_host(h)
         de.assign_sim_host(h, k)
-        for nic in h.nics:  # TODO h.nics does not exist in class HostSim
+        for nic in h.nics:
             de.assign_sim_host(nic, k)
 
             if k != 0:
                 cp.add_nic(nic)
         k = (k + 1) % 2
 
-    for nic in e.nics:  # TODO: e.nics does not exist in class Experiment
+    for nic in e.nics:
         de.add_nic(nic)
 
     return de

experiments/simbricks/simulators.py

@@ -94,6 +94,9 @@ class PCIDevSim(Simulator):
     def full_name(self):
         return 'dev.' + self.name
 
+    def is_nic(self):
+        return False
+
     def sockets_cleanup(self, env):
         return [env.dev_pci_path(self), env.dev_shm_path(self)]
 
@@ -135,6 +138,9 @@ class NICSim(PCIDevSim):
     def full_name(self):
         return 'nic.' + self.name
 
+    def is_nic(self):
+        return True
+
     def sockets_cleanup(self, env):
         return super().sockets_cleanup(env) + [env.nic_eth_path(self)]
 
@@ -215,6 +221,10 @@ class HostSim(Simulator):
         self.pcidevs: tp.List[PCIDevSim] = []
         self.net_directs: tp.List[NetSim] = []
 
+    @property
+    def nics(self):
+        return filter(lambda pcidev: pcidev.is_nic(), self.pcidevs)
+
     def full_name(self):
         return 'host.' + self.name