From 936cdafc79cfc4b467228354f23a2d898129d3f8 Mon Sep 17 00:00:00 2001 From: Your Name Date: Thu, 16 May 2024 18:45:48 +0000 Subject: [PATCH] - --- benchmarks/torchvision/main.py | 2 +- benchmarks/torchvision/voirfile.py | 4 +++- 2 files changed, 4 insertions(+), 2 deletions(-) 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,