pytorch/torch/export
PyTorch MergeBot 1dd6b76914 Revert "[1/N] Remove unused loop variables (#166258)"
This reverts commit 76b2c37045.

Reverted https://github.com/pytorch/pytorch/pull/166258 on behalf of https://github.com/atalman due to breaks test/distributed/test_serialization.py::TestSerialization::test_weights_only [GH job link](https://github.com/pytorch/pytorch/actions/runs/18894311802/job/53929321703) [HUD commit link](76b2c37045) ([comment](https://github.com/pytorch/pytorch/pull/166258#issuecomment-3460964612))
2025-10-29 11:10:37 +00:00
..
experimental Fix pyrefly error syntax (2/n) (#166448) 2025-10-29 00:36:40 +00:00
passes Fix pyrefly error syntax (2/n) (#166448) 2025-10-29 00:36:40 +00:00
pt2_archive Fix pyrefly error syntax (2/n) (#166448) 2025-10-29 00:36:40 +00:00
__init__.py Fix flake8 B028 warnings (#166224) 2025-10-26 06:18:55 +00:00
_draft_export.py Fix pyrefly error syntax (2/n) (#166448) 2025-10-29 00:36:40 +00:00
_leakage_detection_utils.py More ruff SIM fixes (#164695) 2025-10-09 03:24:50 +00:00
_remove_auto_functionalized_pass.py
_remove_effect_tokens_pass.py [BE][1/16] fix typos in torch/ (#156311) 2025-07-09 11:02:22 +00:00
_safeguard.py
_swap.py [pytree] add treespec_{leaf,tuple,dict} functions for args_spec modification (#160843) 2025-10-29 09:16:24 +00:00
_trace.py [pytree] add treespec_{leaf,tuple,dict} functions for args_spec modification (#160843) 2025-10-29 09:16:24 +00:00
_tree_utils.py [pytree] add treespec_{leaf,tuple,dict} functions for args_spec modification (#160843) 2025-10-29 09:16:24 +00:00
_unlift.py [pytree] add treespec_{leaf,tuple,dict} functions for args_spec modification (#160843) 2025-10-29 09:16:24 +00:00
_wrapper_utils.py [reland] Make export._trace._WrapperModule work in strict mode (#146919) (#151264) 2025-04-15 18:35:34 +00:00
custom_obj.py
custom_ops.py Support vmap + custom autograd function/improve DTensor constructor inefficiency (#162240) 2025-09-11 17:42:41 +00:00
decomp_utils.py Back out "Do not decompose in functionalization/proxy tensor if autograd wouldn't have decomposed (#164939)" (#165910) 2025-10-21 16:36:38 +00:00
dynamic_shapes.py Revert "[1/N] Remove unused loop variables (#166258)" 2025-10-29 11:10:37 +00:00
exported_program.py [pytree] add treespec_{leaf,tuple,dict} functions for args_spec modification (#160843) 2025-10-29 09:16:24 +00:00
graph_signature.py Add support for param mutation under inference mode (#159661) 2025-08-14 03:34:04 +00:00
unflatten.py [pytree] add treespec_{leaf,tuple,dict} functions for args_spec modification (#160843) 2025-10-29 09:16:24 +00:00