Revert "Add dont constant fold flag (pytorch#154945)"

This reverts commit 196c95d.

Reverted pytorch#154945 on behalf of https://github.com/malfet due to This broke halide test sanity, see https://hud.pytorch.org/hud/pytorch/pytorch/a3098a74d494020dbb906c05ef047013e1921662/1?per_page=50&name_filter=halide&mergeEphemeralLF=true ([comment](pytorch#154945 (comment)))

Author: pytorchmergebot

test/inductor/test_torchinductor.py

@@ -13550,40 +13550,6 @@ def test_special_polygamma(self):
         self.common(fn, (1, x))
         self.common(fn, (2, x))
 
-    @config.patch({"freezing": True})
-    def test_dont_constant_fold(self):
-        from torch._inductor.constant_folding import (
-            add_dont_constant_fold,
-            clear_dont_constant_fold,
-        )
-
-        m = 5
-
-        class M(torch.nn.Module):
-            def __init__(self):
-                super().__init__()
-                self.w = torch.randn(m)
-                self.s = torch.randn(m)
-
-            def forward(self, x):
-                return self.w * self.s + x
-
-        x = torch.rand(m)
-        mod = M()
-        for dont_constant_fold in [True, False]:
-            clear_dont_constant_fold()
-            if dont_constant_fold:
-                add_dont_constant_fold(torch.ops.aten.mul.Tensor)
-            with torch.no_grad():
-                refe_out = mod(x)
-                mod = torch.compile(mod)
-                test_out, (code,) = run_and_get_code(mod, x)
-            if dont_constant_fold:
-                FileCheck().check("cpp_fused_add_mul").run(code)
-            else:
-                FileCheck().check("cpp_fused_add_0").run(code)
-            self.assertEqual(refe_out, test_out)
-
 
 @dataclasses.dataclass
 class TestFailure:

torch/_inductor/constant_folding.py

@@ -16,18 +16,6 @@
 MODULE_TAG = "_MAIN_MODULE"
 CONST_MODULE_TAG = "_CONST_MODULE"
 
-_dont_constant_fold: list[torch.fx.node.Target] = []
-
-
-def add_dont_constant_fold(op: torch.fx.node.Target) -> None:
-    global _dont_constant_fold
-    _dont_constant_fold.append(op)
-
-
-def clear_dont_constant_fold() -> None:
-    global _dont_constant_fold
-    _dont_constant_fold.clear()
-
 
 def replace_node_with_constant(
     gm: torch.fx.GraphModule,
@@ -158,9 +146,6 @@ def is_woq_int8_pattern(node: torch.fx.node.Node) -> bool:
             # We only folding fp32_weight -> q
             # int8_weight and leave dq in graph to be fused
             return True
-
-        if node.target in _dont_constant_fold:
-            return True
         return False
 
     def node_to_last_non_output_use(self) -> dict[torch.fx.Node, list[torch.fx.Node]]: