diff --git a/torch/export/_draft_export.py b/torch/export/_draft_export.py index 103a4abb054..da9122e751b 100644 --- a/torch/export/_draft_export.py +++ b/torch/export/_draft_export.py @@ -360,7 +360,7 @@ def draft_export( dynamic_shapes: Optional[Union[dict[str, Any], tuple[Any], list[Any]]] = None, preserve_module_call_signature: tuple[str, ...] = (), strict: bool = False, - pre_dispatch: bool = False, + pre_dispatch: bool = True, ) -> ExportedProgram: kwargs = kwargs or {} dynamic_shapes = dynamic_shapes or {}