pytorch/torch/_export
angelayi ea10c072f3 [export] Deserialize args with python keyword names (#136036)
Currently when we deserialize inputs to nodes, we deserialize arguments with default values as kwargs. So deserializing `aten.uniform`, which has the signature `uniform(Tensor(a!) self, float from=0, float to=1, *, Generator? generator=None) -> Tensor(a!)`, will get become `uniform(x, from=0, to=1)`. However, this fails when running in python because `from` is a python keyword. So the solution here is to not deserialize it as a kwarg.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/136036
Approved by: https://github.com/zhxchen17
2024-09-17 18:13:14 +00:00
..
db fix requirement for error classification (#133122) 2024-08-10 04:59:09 +00:00
pass_infra Flip default value for mypy disallow_untyped_defs [1/11] (#127838) 2024-06-08 18:16:33 +00:00
passes [BE]: Update mypy to 1.11.2 (#133816) 2024-09-16 19:44:11 +00:00
serde [export] Deserialize args with python keyword names (#136036) 2024-09-17 18:13:14 +00:00
__init__.py "Remove BLOCK_LIST" (#135729) 2024-09-12 01:22:06 +00:00
converter.py [BE] typing for decorators - fx/_compatibility (part 1) (#134202) 2024-08-22 17:07:33 +00:00
error.py Fix global flake8 issues (#124771) 2024-04-26 15:35:53 +00:00
non_strict_utils.py [Dynamo] Trace torch function modes entered outside of torch.compile (#133137) 2024-09-14 18:52:22 +00:00
pass_base.py [BE] typing for decorators - fx/_compatibility (part 1) (#134202) 2024-08-22 17:07:33 +00:00
tools.py [report_exportability] Avoid re-exporting duplicated modules (#133930) 2024-08-20 22:11:57 +00:00
utils.py [reland][export] fix re-export custom metadata (#135720) 2024-09-13 20:15:15 +00:00
verifier.py [export] Make sure getitem replacement are synced with module call graph. (#134830) 2024-08-30 16:47:05 +00:00
wrappers.py [hop] require hops to override __call__. (#134352) 2024-08-28 19:56:40 +00:00