diff --git a/benchmarks/torchvision/main.py b/benchmarks/torchvision/main.py index 5206666dc..7d6b545ea 100644 --- 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 index 2b0c24d54..2e650bc89 100644 --- 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,