Revert "[dynamo] Record the pre-graph bytecode using fast record function event (pytorch#154769)"

This reverts commit 409c396.

Reverted pytorch#154769 on behalf of https://github.com/seemethere due to This fails internal tests see [fburl.com/diff/67gyp7gp](https://fburl.com/diff/67gyp7gp) ([comment](pytorch#154769 (comment)))

Author: pytorchmergebot

test/inductor/test_compiled_autograd.py

@@ -103,15 +103,10 @@ def reset():
 
 class TestCompiledAutograd(TestCase):
     def setUp(self) -> None:
-        self.exit_stack = contextlib.ExitStack()
-        self.exit_stack.enter_context(
-            config.patch("record_pre_graph_bytecode_in_traces", False)
-        )
         super().setUp()
         reset()
 
     def tearDown(self) -> None:
-        self.exit_stack.close()
         super().tearDown()
         reset()
 

torch/_dynamo/codegen.py

@@ -23,7 +23,7 @@
 import torch.nn
 from torch.utils._ordered_set import OrderedSet
 
-from . import config, graph_break_hints, utils
+from . import graph_break_hints, utils
 from .bytecode_transformation import (
     add_push_null,
     add_push_null_call_function_ex,
@@ -613,18 +613,6 @@ def collect_temp_source(source):
             if arg.source is not None:
                 collect_temp_source(arg.source)
 
-        cm_var = None
-        if config.record_pre_graph_bytecode_in_traces:
-            # Record the pregraph bytecode start
-            self.add_push_null(
-                lambda: self.load_import_from(
-                    utils.__name__, "record_pregraph_bytecode_enter"
-                )
-            )
-            self.extend_output(create_call_function(0, False))
-            cm_var = self.new_var()
-            self.store(cm_var)
-
         for arg in graphargs:
             if arg.pass_arg_as_tensor:
                 self.add_push_null(
@@ -640,18 +628,6 @@ def collect_temp_source(source):
             else:
                 self.call_reconstruct(arg)
 
-        if config.record_pre_graph_bytecode_in_traces:
-            # Record the pregraph bytecode end
-            self.add_push_null(
-                lambda: self.load_import_from(
-                    utils.__name__, "record_pregraph_bytecode_exit"
-                )
-            )
-            assert cm_var is not None
-            self.extend_output([self.create_load(cm_var)])
-            self.extend_output(create_call_function(1, False))
-            self.pop_top()
-
         self.extend_output(create_call_function(len(graphargs), False))
 
     def load_import_from(self, module_name, object_name) -> None:

torch/_dynamo/config.py

@@ -615,9 +615,6 @@ def default_debug_dir_root():
 # wrapper. This ensures that nn.module hooks are also compiled in the same frame.
 wrap_top_frame = False
 
-# record pre-graph bytecode in profile traces
-record_pre_graph_bytecode_in_traces = True
-
 # HACK: this is for testing custom ops profiling only
 _custom_ops_profile: Optional[Any] = None
 

torch/_dynamo/utils.py

@@ -47,7 +47,7 @@
 import warnings
 import weakref
 from collections import Counter, OrderedDict
-from contextlib import AbstractContextManager, contextmanager
+from contextlib import contextmanager
 from dataclasses import is_dataclass
 from functools import lru_cache
 from types import MethodWrapperType
@@ -4670,17 +4670,3 @@ def maybe_disable_inference_mode_for_fake_prop() -> Generator[None, None, None]:
 
 def is_node_meta_valid(node: Optional[torch.fx.Node]) -> bool:
     return node is None or "example_value" in node.meta or "val" in node.meta
-
-
-def record_pregraph_bytecode_enter() -> AbstractContextManager[None]:
-    cm: AbstractContextManager[None] = (
-        torch._C._profiler._RecordFunctionFast("Pregraph bytecode")
-        if torch.autograd.profiler._is_profiler_enabled
-        else contextlib.nullcontext()
-    )
-    cm.__enter__()
-    return cm
-
-
-def record_pregraph_bytecode_exit(cm: AbstractContextManager[None]) -> None:
-    cm.__exit__(None, None, None)