turn off normalize_ir, turn on use_functionalization by default

ghstack-source-id: 1561bb9
Pull Request resolved: #1026

Author: bdhirsh

torchdynamo/__init__.py

@@ -13,6 +13,11 @@
 from .utils import guard_failures
 from .utils import orig_code_map
 
+# TODO: remove this config entirely
+import functorch.compile
+
+functorch.compile.config.use_functionalize = True
+
 __all__ = [
     "optimize",
     "optimize_assert",

torchdynamo/config.py

@@ -66,7 +66,7 @@ class AccessLimitingConfig(ModuleType):
     fake_tensor_propagation = True
 
     # run FX normalization passes in optimizer
-    normalize_ir = True
+    normalize_ir = False
 
     # If a tensor subclass type is in this set, torchdynamo will inline the
     # __torch_function__ logic of the subclass.