added simualtor cleanup_commands

experiments/simbricks/orchestration/simulation/base.py

@@ -249,6 +249,9 @@ class Simulator(utils_base.IdObj):
     def checkpoint_commands(self) -> list[str]:
         return []
     
+    def cleanup_commands(self) -> list[str]:
+        return []
+
     @abc.abstractmethod
     def supported_socket_types(self) -> set[inst_base.SockType]:
         return []

experiments/simbricks/orchestration/simulation/host.py

@@ -91,6 +91,9 @@ class Gem5Sim(HostSim):
     def checkpoint_commands(self) -> list[str]:
         return ["m5 checkpoint"]
     
+    def cleanup_commands(self) -> list[str]:
+        return ["m5 exit"]
+
     def run_cmd(self, inst: inst_base.Instantiation) -> str:
         cpu_type = self.cpu_type
         if inst.create_cp():

experiments/simbricks/orchestration/system/host/base.py

@@ -105,9 +105,13 @@ class BaseLinuxHost(FullSystemHost):
 
     def cleanup_cmds(self, inst: instantiation.Instantiation) -> list[str]:
         """Commands to run to cleanup node."""
-        return self._concat_app_cmds(
+        cmds = self._concat_app_cmds(
             inst, app.BaseLinuxApplication.cleanup_cmds.__name__
         ) 
+        sim = inst.find_sim_by_spec(spec=self)
+        cleanup = sim.cleanup_commands()
+        cmds += cleanup
+        return cmds
 
     def config_files(self, inst: instantiation.Instantiation) -> dict[str, tp.IO]:
         """