Prohibit customize export config

Differential Revision: D49258936

Author: guangy10

examples/export/utils.py

@@ -27,14 +27,13 @@
 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, _CAPTURE_CONFIG)
     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,
 ):