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/export/utils.py

@@ -16,8 +16,6 @@
 from executorch.exir.tracer import Value
 
 
-_CAPTURE_CONFIG = exir.CaptureConfig(enable_aot=True)
-
 # Explicitly force the activation of the IR validator
 _EDGE_COMPILE_CONFIG = exir.EdgeCompileConfig(
     _check_ir_validity=True,
@@ -27,14 +25,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 +50,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,
 ):