diff --git a/userbenchmark/dynamo/dynamobench/_dynamo/utils.py b/userbenchmark/dynamo/dynamobench/_dynamo/utils.py index 6b4acbe1b0..3eaba129af 100644 --- a/userbenchmark/dynamo/dynamobench/_dynamo/utils.py +++ b/userbenchmark/dynamo/dynamobench/_dynamo/utils.py @@ -255,10 +255,10 @@ def time_wrapper(*args, **kwargs): frame_key = str(curr_frame) if frame_key not in frame_phase_timing: frame_phase_timing[frame_key] = {} - assert ( - phase_name not in frame_phase_timing[frame_key] - ), f"Duplicate phase name {phase_name} for frame {frame_key}" - frame_phase_timing[frame_key][phase_name] = time_spent + if phase_name not in frame_phase_timing[frame_key]: + frame_phase_timing[frame_key][phase_name] = time_spent + else: + frame_phase_timing[frame_key][phase_name] += time_spent return r return time_wrapper @@ -621,6 +621,8 @@ class CompilationMetrics: graph_input_count: Optional[int] entire_frame_compile_time_s: Optional[float] backend_compile_time_s: Optional[float] + inductor_compile_time_s: Optional[float] + code_gen_time_s: Optional[float] fail_type: Optional[str] fail_reason: Optional[str] fail_user_frame_filename: Optional[str]