Update opt to use stdout

benchmarks/accelerate_opt/benchfile.py

@@ -35,8 +35,14 @@ def build_prepare_plan(self):
             str(self.dirs.cache)
         )
 
-    def build_run_plan(self):
-        plan = super().build_run_plan()
+    def build_run_plan(self): 
+        from milabench.commands import PackCommand, VoirCommand
+        main = self.dirs.code / self.main_script
+        plan = PackCommand(self, *self.argv, lazy=True)
+
+        if False:
+            plan = VoirCommand(pack, cwd=main.parent)
+
         return AccelerateAllNodes(plan).use_stdout()
 
 

benchmarks/accelerate_opt/voirfile.py

@@ -0,0 +1,34 @@
+from dataclasses import dataclass
+
+from voir import configurable
+from voir.phase import StopProgram
+
+
+@dataclass
+class Config:
+    """voir configuration"""
+
+    # Whether to display the dash or not
+    dash: bool = False
+
+    # How often to log the rates
+    interval: str = "1s"
+
+    # Number of rates to skip before logging
+    skip: int = 5
+
+    # Number of rates to log before stopping
+    stop: int = 20
+
+    # Number of seconds between each gpu poll
+    gpu_poll: int = 3
+
+
+@configurable
+def instrument_main(ov, options: Config):
+    yield ov.phases.init
+
+    try:
+        yield ov.phases.run_script
+    except StopProgram:
+        print("early stopped")

benchmarks/lightning/voirfile.py

@@ -0,0 +1,34 @@
+from dataclasses import dataclass
+
+from voir import configurable
+from voir.phase import StopProgram
+
+
+@dataclass
+class Config:
+    """voir configuration"""
+
+    # Whether to display the dash or not
+    dash: bool = False
+
+    # How often to log the rates
+    interval: str = "1s"
+
+    # Number of rates to skip before logging
+    skip: int = 5
+
+    # Number of rates to log before stopping
+    stop: int = 20
+
+    # Number of seconds between each gpu poll
+    gpu_poll: int = 3
+
+
+@configurable
+def instrument_main(ov, options: Config):
+    yield ov.phases.init
+
+    try:
+        yield ov.phases.run_script
+    except StopProgram:
+        print("early stopped")

benchmarks/torchvision_ddp/voirfile.py

@@ -1,5 +1,9 @@
 from voir.phase import StopProgram
 
+from voir import configurable
+from voir.phase import StopProgram
+
+
 
 @configurable
 def instrument_main(ov):