pytorch/torch/distributed/algorithms
Edward Z. Yang 46712b019d Enable local_partial_types (#118467)
When using dmypy, this setting is enabled and cannot be turned off. Force it for regular mypy too.

Signed-off-by: Edward Z. Yang <ezyang@meta.com>

Pull Request resolved: https://github.com/pytorch/pytorch/pull/118467
Approved by: https://github.com/Skylion007
ghstack dependencies: #118414, #118418, #118432
2024-01-28 13:38:22 +00:00
..
_checkpoint Fix docstring errors in default_hooks.py, optimizer_overlap.py, checkpoint_wrapper.py, copy.py, benchmark_ddp_rpc.py, utils.py, dependency.py, phony.py, pipeline.py, checkpoint.py, worker.py, batchnorm.py, quantization.py (#113511) 2023-11-14 18:52:41 +00:00
_comm_hooks Fix docstring errors in default_hooks.py, optimizer_overlap.py, checkpoint_wrapper.py, copy.py, benchmark_ddp_rpc.py, utils.py, dependency.py, phony.py, pipeline.py, checkpoint.py, worker.py, batchnorm.py, quantization.py (#113511) 2023-11-14 18:52:41 +00:00
_optimizer_overlap Fix docstring errors in default_hooks.py, optimizer_overlap.py, checkpoint_wrapper.py, copy.py, benchmark_ddp_rpc.py, utils.py, dependency.py, phony.py, pipeline.py, checkpoint.py, worker.py, batchnorm.py, quantization.py (#113511) 2023-11-14 18:52:41 +00:00
_quantization Fix docstring errors in default_hooks.py, optimizer_overlap.py, checkpoint_wrapper.py, copy.py, benchmark_ddp_rpc.py, utils.py, dependency.py, phony.py, pipeline.py, checkpoint.py, worker.py, batchnorm.py, quantization.py (#113511) 2023-11-14 18:52:41 +00:00
ddp_comm_hooks Enable local_partial_types (#118467) 2024-01-28 13:38:22 +00:00
model_averaging Docs: fix docstring errors in model_averaging (#117038) 2024-01-18 04:12:51 +00:00
__init__.py
join.py fix docstring issues in torch.distributed (#113337) 2023-11-13 19:37:29 +00:00