Revert direct updates (#159654)

izaitsevfb · pytorchmergebot · commit e4b123b5e4dd · 2025-08-01T16:54:51.000Z
reverts: ``` commit 5711a8f (tag: trunk/5711a8f06948eeee56ed5f53f171fa519f78491c, origin/main, main) Author: Jovian Anthony Jaison <38627145+jovianjaison@users.noreply.github.com> Date: Fri Aug 1 09:32:52 2025 -0700 Update test_utils.py commit b4b71d0 (tag: trunk/b4b71d011ed07a41c2086ff0dec2988a63662877) Author: Jovian Anthony Jaison <38627145+jovianjaison@users.noreply.github.com> Date: Fri Aug 1 09:27:54 2025 -0700 Update utils.py commit 52376b9 (tag: trunk/52376b9b6fbf9fe24f5d82038dc520f0c64b6f8d) Author: Jovian Anthony Jaison <38627145+jovianjaison@users.noreply.github.com> Date: Fri Aug 1 09:26:05 2025 -0700 ``` (commits pushed directly to main by mistake) Pull Request resolved: #159654 Approved by: https://github.com/atalman
diff --git a/test/dynamo/test_utils.py b/test/dynamo/test_utils.py
@@ -350,7 +350,6 @@ def test_dynamo_timed(self, mock_time, mock_time_ns):
             e.cuda_version = None
             e.triton_version = None
             e.python_version = None
-            e.stack_trace = None
 
         # First event is for the forward. Formatting makes reading diffs
         # much easier.
@@ -434,7 +433,6 @@ def test_dynamo_timed(self, mock_time, mock_time_ns):
  'runtime_triton_autotune_time_us': None,
  'shape_env_guard_count': 0,
  'specialize_float': False,
- 'stack_trace': None,
  'start_time': 0.0001,
  'start_time_us': 100,
  'structured_logging_overhead_s': 0.0,
@@ -527,7 +525,6 @@ def test_dynamo_timed(self, mock_time, mock_time_ns):
  'runtime_triton_autotune_time_us': None,
  'shape_env_guard_count': None,
  'specialize_float': None,
- 'stack_trace': None,
  'start_time': 0.0001,
  'start_time_us': 100,
  'structured_logging_overhead_s': 0.0,
diff --git a/torch/_dynamo/convert_frame.py b/torch/_dynamo/convert_frame.py
@@ -1300,7 +1300,6 @@ def format_func_info(code: CodeType) -> str:
                 "dynamo_compile_time_before_restart_us": to_int_us(
                     dynamo_time_before_restart
                 ),
-                "stack_trace": traceback.format_stack(),
             }
             # TODO: replace with CompileEventLogger.compilation_metrics
             # There are some columns here not in PT2 Compile Events
diff --git a/torch/_dynamo/utils.py b/torch/_dynamo/utils.py
@@ -1279,7 +1279,6 @@ class CompilationMetrics:
     compliant_custom_ops: Optional[set[str]] = None
     restart_reasons: Optional[set[str]] = None
     dynamo_time_before_restart_s: Optional[float] = None
-    stack_trace: Optional[list[str]] = None
     # Sometimes, we will finish analyzing a frame but conclude we don't want
     # to install any guarded code.  True means we actually decided to install
     # a compiled frame
@@ -1475,7 +1474,6 @@ def add_compilation_metrics_to_chromium(c: CompilationMetrics) -> None:
         fail_reason=c.fail_reason,
         fail_user_frame_filename=c.fail_user_frame_filename,
         fail_user_frame_lineno=c.fail_user_frame_lineno,
-        stack_trace=c.stack_trace,
         # Sets aren't JSON serializable
         non_compliant_ops=list(c.non_compliant_ops)
         if c.non_compliant_ops is not None

Original file line number	Diff line number	Diff line change
`@@ -1300,7 +1300,6 @@ def format_func_info(code: CodeType) -> str:`
`1300`	`1300`	`"dynamo_compile_time_before_restart_us": to_int_us(`
`1301`	`1301`	`dynamo_time_before_restart`
`1302`	`1302`	`),`
`1303`		`- "stack_trace": traceback.format_stack(),`
`1304`	`1303`	`}`
`1305`	`1304`	`# TODO: replace with CompileEventLogger.compilation_metrics`
`1306`	`1305`	`# There are some columns here not in PT2 Compile Events`