Fix issues with guard serialization on distributed types. (#162418)

zhxchen17 · facebook-github-bot · commit f42b77a79973 · 2025-09-12T09:09:10.000-07:00
Summary: Add more support for torch internal distributed data structures. X-link: pytorch/pytorch#162418 Approved by: https://github.com/dolpm Reviewed By: dolpm Differential Revision: D81927732 fbshipit-source-id: 50712f0052dff74332a8db3efe9f553b110bc53c
diff --git a/userbenchmark/dynamo/dynamobench/_dynamo/utils.py b/userbenchmark/dynamo/dynamobench/_dynamo/utils.py
@@ -709,6 +709,7 @@ def _foo(...):
     if key not in compilation_time_metrics:
         compilation_time_metrics[key] = []
 
+    metrics = compilation_time_metrics[key]
     event_metadata = {}
     if metadata:
         event_metadata.update(metadata)
@@ -756,7 +757,7 @@ def _foo(...):
     finally:
         end_ns = time.time_ns()
         time_spent_ns = end_ns - start_ns
-        compilation_time_metrics[key].append(time_spent_ns / 1e9)
+        metrics.append(time_spent_ns / 1e9)
         chromium_log.log_event_end(
             event_name, end_ns, {}, start_ns, log_pt2_compile_event, compile_id
         )
