diff --git a/run.py b/run.py index 54bd9073bf..01dda5d624 100644 --- a/run.py +++ b/run.py @@ -31,7 +31,9 @@ if not hasattr(torch.version, "git_version"): - from pytorch.benchmark.fb.run_utils import trace_handler + from pytorch.benchmark.fb.run_utils import trace_handler, usage_report_logger +else: + usage_report_logger = lambda: None WARMUP_ROUNDS = 3 @@ -375,6 +377,9 @@ def _validate_profile_options(profile_options: str): print("cuda device required to use --cudastreams option!") exit(-1) + # Log the tool usage + usage_report_logger() + found = False Model = None