diff --git a/userbenchmark/dynamo/dynamobench/_dynamo/utils.py b/userbenchmark/dynamo/dynamobench/_dynamo/utils.py index a41d92724e..b66b17a98c 100644 --- a/userbenchmark/dynamo/dynamobench/_dynamo/utils.py +++ b/userbenchmark/dynamo/dynamobench/_dynamo/utils.py @@ -774,6 +774,7 @@ def proxy_args_kwargs(args, kwargs): @dataclasses.dataclass class CompilationMetrics: + is_forward: bool = dataclasses.field(default=True, init=False) compile_id: str frame_key: str co_name: str @@ -814,6 +815,7 @@ class CompilationMetrics: @dataclasses.dataclass class BwdCompilationMetrics: + is_forward: bool = dataclasses.field(default=False, init=False) compile_id: str inductor_compile_time_s: Optional[float] code_gen_time_s: Optional[float]