Skip to content

Commit 98b06f0

Browse files
masnesralfacebook-github-bot
authored andcommitted
Add pgo remote get/put timings to dynamo_compile (#150322)
Summary: X-link: pytorch/pytorch#150322 Approved by: https://github.com/ppanchalia Reviewed By: atalman Differential Revision: D72597454 fbshipit-source-id: 07e20f9819910f1b6d3339718fb6722d41f5e67d
1 parent de0cb2f commit 98b06f0

File tree

1 file changed

+2
-0
lines changed
  • userbenchmark/dynamo/dynamobench/_dynamo

1 file changed

+2
-0
lines changed

userbenchmark/dynamo/dynamobench/_dynamo/utils.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1267,6 +1267,8 @@ class CompilationMetrics:
12671267
ir_count: Optional[int] = None
12681268
cudagraph_skip_reason: Optional[str] = None
12691269
python_version: Optional[str] = None
1270+
pgo_put_remote_code_state_time_us: Optional[int] = None
1271+
pgo_get_remote_code_state_time_us: Optional[int] = None
12701272

12711273
@classmethod
12721274
def create(cls, metrics: dict[str, Any]):

0 commit comments

Comments
 (0)