frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add type annotations for torch._C._distributed_c10d module. (#46623)

Browse Source 
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/46623

Test Plan: Imported from OSS

Reviewed By: malfet

Differential Revision: D24761606

Pulled By: xuzhao9

fbshipit-source-id: 827eaf2502e381ee24d36741c1613b4c08208569
This commit is contained in:
Xu Zhao
2020-11-06 00:47:23 -08:00
committed by Facebook GitHub Bot
parent fe77ded48a
commit 73a3e70b24
5 changed files with 391 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
torch/distributed/rendezvous.py
View File

@ -7,11 +7,11 @@ import torch._six as six
import numbers
import os
import sys
from . import FileStore
from torch._C._distributed_c10d import FileStore
from .constants import default_pg_timeout
if sys.platform != 'win32':
from . import TCPStore
from torch._C._distributed_c10d import TCPStore
_rendezvous_handlers = {}