pytorch

2cc624cd - Enable mypy check for distributed.py, and fix type errors (#87543)

Enable mypy check for distributed.py, and fix type errors (#87543) Pull Request resolved: https://github.com/pytorch/pytorch/pull/87543 Approved by: https://github.com/fduwjj