Enable bwd compilation metrics (#128973)

Summary:
Fixes #ISSUE_NUMBER

X-link: pytorch/pytorch#128973
Approved by: https://github.com/dshi7

Reviewed By: fbgheith

Differential Revision: D58798131

Pulled By: yanboliang

fbshipit-source-id: 03c5b35f5dbdfa16c54300e22daa902d5be904fa

userbenchmark/dynamo/dynamobench/_dynamo/utils.py

@@ -714,18 +714,15 @@ def record_compilation_metrics(
  name = "compilation_metrics"
  else:
  name = "bwd_compilation_metrics"
- # Currently only record fwd compilation metrics, will add bwd compilation metrics
- # after the internal Scuba logging changes finish.
- if isinstance(compilation_metrics, CompilationMetrics):
- torch._logging.trace_structured(
- name,
- lambda: {
- k: list(v) if isinstance(v, set) else v
- for k, v in dataclasses.asdict(compilation_metrics).items()
- },
- )
- if config.log_compilation_metrics:
- log_compilation_event(compilation_metrics)
+ torch._logging.trace_structured(
+ name,
+ lambda: {
+ k: list(v) if isinstance(v, set) else v
+ for k, v in dataclasses.asdict(compilation_metrics).items()
+ },
+ )
+ if config.log_compilation_metrics:
+ log_compilation_event(compilation_metrics)
 
 
 def set_compilation_metrics_limit(new_size: int) -> None: