Skip to content

Commit

Permalink
Add start event metadata to collected metadata for PT2 Compile Events
Browse files Browse the repository at this point in the history
Summary:
X-link: pytorch/pytorch#139289

We should be logging metadata from event starts to PT2 Compile Events too.
ghstack-source-id: 250444771

Reviewed By: oulgen

Differential Revision: D65070086

fbshipit-source-id: 63b934bff4254871e15a615e5aa47112b032b143
  • Loading branch information
jamesjwu authored and facebook-github-bot committed Oct 31, 2024
1 parent a0890b0 commit 0c8a0f6
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions userbenchmark/dynamo/dynamobench/_dynamo/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -1023,6 +1023,8 @@ def log_event_start(
metadata,
)
self.get_stack().append(event_name)
# Add metadata from start event
self.add_event_data(event_name, **metadata)

def reset(self) -> None:
# We this on every compile in case a compile crashes or restarts and we haven't
Expand Down

0 comments on commit 0c8a0f6

Please sign in to comment.