Revert "[dynamo] Avoid running torch.nn.Module.__call__ twice under torch.compile(mod) (#152740)"

pytorchmergebot · pytorchmergebot · commit d36261d2e65f · 2025-05-08T17:31:21.000Z
This reverts commit 0886d40. Reverted #152740 on behalf of https://github.com/huydhn due to Discuss with the author to revert and reland this ([comment](#152740 (comment)))
diff --git a/test/dynamo/test_hooks.py b/test/dynamo/test_hooks.py
@@ -889,32 +889,6 @@ def forward(self, x):
         self.assertEqual(ref, res)
         self.assertEqual(cnts.frame_count, 1)
 
-    def test_global_module_forward_pre_hook(self):
-        class Mod(torch.nn.Module):
-            def forward(self, x):
-                return x - 1
-
-        counter = 0
-
-        def hook(mod, args):
-            nonlocal counter
-            counter += 1
-            return (args[0] + 100,)
-
-        mod = Mod()
-        torch.nn.modules.module.register_module_forward_pre_hook(hook)
-
-        # Case 1: torch.compile(mod)
-        compiled_mod = torch.compile(mod, backend="eager")
-
-        x = torch.rand(18, 18)
-
-        ref = mod(x)
-        self.assertEqual(counter, 1)
-        res = compiled_mod(x)
-        self.assertEqual(counter, 2)
-        self.assertEqual(ref, res)
-
 
 if __name__ == "__main__":
     from torch._dynamo.test_case import run_tests
diff --git a/torch/_dynamo/eval_frame.py b/torch/_dynamo/eval_frame.py
@@ -343,12 +343,6 @@ def _initialize(self):
             self._forward = self.forward
             self.forward = self._call_lazy_check
 
-    def __call__(self, *args, **kwargs):
-        # All the logic in `torch.nn.Module.__call__` has been captured by
-        # `self.forward = self.dynamo_ctx(self._orig_mod.__call__)`, so we
-        # override here to avoid running that logic again by default.
-        return self.forward(*args, **kwargs)
-
     def __reduce__(self):
         return (self.__class__, (self._orig_mod, self.dynamo_ctx))
 
