Revert "[Dynamo] Allow inlining into AO quantization modules (pytorch#152934)" (pytorch#158677)

This reverts commit 20e2ca3.

Author: atalman

test/dynamo/test_repros.py

@@ -6450,21 +6450,6 @@ def inject_parameters(module, cls):
         with torch.no_grad():
             model(x)
 
-    def test_ao_fake_quantize_tracing(self):
-        import torch.ao.quantization.fake_quantize
-
-        q = torch.ao.quantization.FusedMovingAvgObsFakeQuantize()
-
-        def fn(x):
-            return q(x)
-
-        x = torch.ones(2, 2)
-        opt_fn = torch.compile(fn, backend="eager", fullgraph=True)
-        res = opt_fn(x)
-        eager_res = fn(x)
-
-        self.assertEqual(res, eager_res)
-
     def test_typed_dict(self):
         class LlavaImagePixelInputs(TypedDict):
             type: Literal["pixel_values"]

torch/_dynamo/trace_rules.py

@@ -3393,7 +3393,6 @@ def _module_dir(m: types.ModuleType):
     "torch._tensor",
     "torch.amp.autocast_mode",
     "torch.ao.nn",
-    "torch.ao.quantization.fake_quantize",
     "torch.autograd.function",
     "torch.backends.cuda",
     "torch.cuda.amp.autocast_mode",