Prohibit customize export config (#337)

Summary:

More cleanup in `executorch/examples`.

`_unlift` flag is removed for all models and all use-cases in examples, we can push to use canonical export config one step further.

Not in this diff but `enable_aot` will be removed later together with `exir.capture()`

Reviewed By: digantdesai

Differential Revision: D49258936

Author: guangy10

examples/backend/xnnpack_examples.py

@@ -12,7 +12,7 @@
 import torch._export as export
 
 from executorch.backends.xnnpack.partition.xnnpack_partitioner import XnnpackPartitioner
-from executorch.exir import CaptureConfig, EdgeCompileConfig
+from executorch.exir import EdgeCompileConfig
 from executorch.exir.backend.backend_api import to_backend
 
 from ..export.utils import export_to_edge, save_pte_program
@@ -82,7 +82,6 @@
     edge = export_to_edge(
         model,
         example_inputs,
-        capture_config=CaptureConfig(enable_aot=True),
         edge_compile_config=EdgeCompileConfig(
             _check_ir_validity=False if args.quantize else True,
         ),

examples/export/utils.py

@@ -16,9 +16,6 @@
 from executorch.exir.tracer import Value
 
 
-_CAPTURE_CONFIG = exir.CaptureConfig(enable_aot=True)
-
-# TODO(T163721729): Enable IR check after decomposing div.Tensor_mode
 _EDGE_COMPILE_CONFIG = exir.EdgeCompileConfig(
     _check_ir_validity=False,
 )
@@ -27,14 +24,15 @@
 def _to_core_aten(
     model: torch.fx.GraphModule,
     example_inputs: Tuple[Value, ...],
-    capture_config=_CAPTURE_CONFIG,
 ) -> ExirExportedProgram:
     # post autograd export. eventually this will become .to_core_aten
     if not isinstance(model, torch.fx.GraphModule):
         raise ValueError(
             f"Expected passed in model to be an instance of fx.GraphModule, got {type(model)}"
         )
-    core_aten_exir_ep = exir.capture(model, example_inputs, capture_config)
+    core_aten_exir_ep = exir.capture(
+        model, example_inputs, exir.CaptureConfig(enable_aot=True)
+    )
     logging.info(f"Core ATen graph:\n{core_aten_exir_ep.exported_program.graph}")
     return core_aten_exir_ep
 
@@ -51,17 +49,15 @@ def _core_aten_to_edge(
 def export_to_edge(
     model: torch.fx.GraphModule,
     example_inputs: Tuple[Value, ...],
-    capture_config=_CAPTURE_CONFIG,
     edge_compile_config=_EDGE_COMPILE_CONFIG,
 ) -> ExirExportedProgram:
-    core_aten_exir_ep = _to_core_aten(model, example_inputs, capture_config)
+    core_aten_exir_ep = _to_core_aten(model, example_inputs)
     return _core_aten_to_edge(core_aten_exir_ep, edge_compile_config)
 
 
 def export_to_exec_prog(
     model,
     example_inputs,
-    capture_config=_CAPTURE_CONFIG,
     edge_compile_config=_EDGE_COMPILE_CONFIG,
     backend_config=None,
 ):