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
c17ba69ba5
pytorch/torch/export
History
Fabian Keller 394c339691 improve typings in unflatten (#141817) 
A first follow-up to https://github.com/pytorch/pytorch/pull/115074 / https://github.com/pytorch/pytorch/pull/141240 following the strategy discussed there (https://github.com/pytorch/pytorch/pull/115074#issuecomment-2480992230).

This PR improves the type annotations around `unflatten.py` which had been inaccurate due to the previously suppressed type checking on `torch.nn.Module`.

CC @Skylion007 @ezyang

Pull Request resolved: https://github.com/pytorch/pytorch/pull/141817
Approved by: https://github.com/Skylion007
2024-11-30 22:12:15 +00:00
..
experimental Fix type-safety of torch.nn.Module instances (#141240) 2024-11-22 00:05:05 +00:00
passes Fixing issue in move pass for copying Parameter (#138855) 2024-10-25 17:57:27 +00:00
__init__.py Replace torch.export default decomp table to be lazily populated (#137650) 2024-10-18 19:28:52 +00:00
_draft_export.py [export] Improve stacktrace filtering (#141285) 2024-11-22 20:55:04 +00:00
_remove_auto_functionalized_pass.py Redesign custom op functionlaization for better re-inplace (#134409) 2024-09-04 17:08:58 +00:00
_remove_effect_tokens_pass.py [export][fx] More robust DCE pass (#132764) 2024-08-06 22:27:22 +00:00
_safeguard.py Flip default value for mypy disallow_untyped_defs [6/11] (#127843) 2024-06-08 18:49:29 +00:00
_swap.py Fix type-safety of torch.nn.Module instances (#141240) 2024-11-22 00:05:05 +00:00
_trace.py [BE] Make maybe_aliasing_or_mutating proper tag (#131990) 2024-11-24 00:12:49 +00:00
_tree_utils.py [export] Add setattr for ep.example_inputs (#140990) 2024-11-20 02:49:20 +00:00
_unlift.py [export] Add setattr for ep.example_inputs (#140990) 2024-11-20 02:49:20 +00:00
custom_obj.py [Export] Introduce class_fqn into CustomObjArgument (#118158) 2024-01-25 18:44:25 +00:00
decomp_utils.py Replace torch.export default decomp table to be lazily populated (#137650) 2024-10-18 19:28:52 +00:00
dynamic_shapes.py make DimHints compatible with Dims (#138490) 2024-10-22 07:43:48 +00:00
exported_program.py Revert "Add truediv support in export serializer (#136364)" 2024-11-27 03:24:31 +00:00
graph_signature.py Revert "Add truediv support in export serializer (#136364)" 2024-11-27 03:24:31 +00:00
unflatten.py improve typings in unflatten (#141817) 2024-11-30 22:12:15 +00:00