OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 00:21:07 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
71cb13869b
pytorch/torch/distributed/optim
History
Jez Ng 631fb33fd6 Enable import following in MYPYNOFOLLOW (now MYPYINDUCTOR) (#113830) 
Skipping importing some packages for now to make this change more
tractable.

For some reason, lintrunner on CI raises errors in all imported `.pyi` files,
even though it doesn't on my local machine. The errors are all from missing
generic types, as the MYPYINDUCTOR config has `disallow_any_generics`
set. I have thus added `disable-error-code` comments to the relevant files,
though I fixed a few that were easy enough.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/113830
Approved by: https://github.com/Skylion007
ghstack dependencies: #113722, #113721
2023-11-17 18:24:21 +00:00
..
__init__.py
apply_optimizer_in_backward.py [Distributed] Small nits to apply_optimizer_in_backward (#110903) 2023-10-11 07:45:45 +00:00
functional_adadelta.py
functional_adagrad.py [optim/dynamo] shortcut adagrad with has_complex (#112722) 2023-11-02 16:50:45 +00:00
functional_adam.py perf(inductor): improve Adam compile times by shortcutting for loops (via has_complex) (#110607) 2023-10-06 05:08:49 +00:00
functional_adamax.py feat(optim): use has_complex shortcut flag for all applicable optimizers, use _view_as_real auxiliary function (#110706) 2023-10-31 20:33:03 +00:00
functional_adamw.py feat(optim): use has_complex shortcut flag for all applicable optimizers, use _view_as_real auxiliary function (#110706) 2023-10-31 20:33:03 +00:00
functional_rmsprop.py feat(optim): use has_complex shortcut flag for all applicable optimizers, use _view_as_real auxiliary function (#110706) 2023-10-31 20:33:03 +00:00
functional_rprop.py feat(optim): use has_complex shortcut flag for all applicable optimizers, use _view_as_real auxiliary function (#110706) 2023-10-31 20:33:03 +00:00
functional_sgd.py
named_optimizer.py fix docstring issues in torch.distributed (#113337) 2023-11-13 19:37:29 +00:00
optimizer.py
post_localSGD_optimizer.py
utils.py
zero_redundancy_optimizer.py Enable import following in MYPYNOFOLLOW (now MYPYINDUCTOR) (#113830) 2023-11-17 18:24:21 +00:00
zero_redundancy_optimizer.pyi Enable import following in MYPYNOFOLLOW (now MYPYINDUCTOR) (#113830) 2023-11-17 18:24:21 +00:00