OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

[export] Make draft-export predispatch=True by default (#151051)

Browse Source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/151051
Approved by: https://github.com/pianpwk
This commit is contained in:
angelayi 2025-04-15 11:18:46 -07:00 committed by PyTorch MergeBot
parent a5c61668d7
commit 84e633e09d
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
torch/export/_draft_export.py
View File

@ -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 {}