Skip to content

Commit e89c1b3

Browse files
masnesralfacebook-github-bot
authored andcommitted
Log is_forward field to dynamo_compile scuba table (#2511)
Summary: Pull Request resolved: #2511 X-link: pytorch/pytorch#138097 ^^ Reviewed By: ezyang Differential Revision: D64438144 fbshipit-source-id: 87a5518d4d9318132d269302c93a285bf86f3a46
1 parent 04f0e6c commit e89c1b3

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
@@ -774,6 +774,7 @@ def proxy_args_kwargs(args, kwargs):
774774

775775
@dataclasses.dataclass
776776
class CompilationMetrics:
777+
is_forward: bool = dataclasses.field(default=True, init=False)
777778
compile_id: str
778779
frame_key: str
779780
co_name: str
@@ -814,6 +815,7 @@ class CompilationMetrics:
814815

815816
@dataclasses.dataclass
816817
class BwdCompilationMetrics:
818+
is_forward: bool = dataclasses.field(default=False, init=False)
817819
compile_id: str
818820
inductor_compile_time_s: Optional[float]
819821
code_gen_time_s: Optional[float]

0 commit comments

Comments
 (0)