-

mila-iqia · May 16, 2024 · 936cdaf · 936cdaf
1 parent 009a857
commit 936cdaf
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/benchmarks/torchvision/main.py b/benchmarks/torchvision/main.py
@@ -11,7 +11,7 @@
 import torchcompat.core as accelerator
 
 import voir
-from voir.asynctimer import DataloaderWrapper, DataloaderWrapperGiver, DataloaderWrapperSmuggle, StopProgram
+from voir.wrapper import DataloaderWrapper, DataloaderWrapperGiver, DataloaderWrapperSmuggle, StopProgram
 from giving import give, given
 from cantilever.core.timer import timeit, timeiterator, show_timings
 

diff --git a/benchmarks/torchvision/voirfile.py b/benchmarks/torchvision/voirfile.py
@@ -33,8 +33,10 @@ def instrument_main(ov, options: Config):
     if options.dash:
         ov.require(dash)
 
+    overhead_metrics = [] # "__iter__", "overhead", "process_time"
+
     ov.require(
-        log("value", "progress", "rate", "units", "loss", "gpudata", context="task"),
+        log("value", "progress", "rate", "units", "loss", "gpudata", *overhead_metrics, context="task"),
         rate(
             interval=options.interval,
             skip=options.skip,