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: