diff --git a/userbenchmark/dynamo/dynamobench/_dynamo/utils.py b/userbenchmark/dynamo/dynamobench/_dynamo/utils.py index 65095e7da..f0b736540 100644 --- a/userbenchmark/dynamo/dynamobench/_dynamo/utils.py +++ b/userbenchmark/dynamo/dynamobench/_dynamo/utils.py @@ -76,6 +76,7 @@ from torch._utils_internal import ( log_chromium_event_internal, log_compilation_event, + record_chromium_event_internal, signpost_event, ) from torch.fx._utils import _format_graph_code, lazy_format_graph_code @@ -1175,6 +1176,7 @@ def _log_timed_event( suppress_context=False, expect_trace_id=False, # Not every chromium event will have a trace_id ) + record_chromium_event_internal(event) return event def log_instant_event(