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

Revert D23841786: [pytorch][PR] Enable distributed package on windows, Gloo backend supported only

Browse Source 
Test Plan: revert-hammer

Differential Revision:
D23841786 (0122299f9b)

Original commit changeset: 334ba1ed73ef

fbshipit-source-id: ec95432f9957df56a5a04e52661f5db920b7f57f
This commit is contained in:
Mike Ruberry
2020-09-24 22:42:46 -07:00
committed by Facebook GitHub Bot
parent bc3151dee0
commit 103fa3894a
39 changed files with 167 additions and 462 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
torch/distributed/rendezvous.py
View File

@ -6,12 +6,9 @@ except ImportError:
import torch._six as six
import numbers
import os
import sys
from . import FileStore
from . import FileStore, TCPStore
from .constants import default_pg_timeout
if sys.platform != 'win32':
from . import TCPStore
_rendezvous_handlers = {}
@ -93,10 +90,6 @@ def _file_rendezvous_handler(url, **kwargs):
result = urlparse(url)
path = result.path
if sys.platform == 'win32':
import urllib.request
path = urllib.request.url2pathname(result.path)
if not path:
raise _error("path missing")
query = dict(pair.split("=") for pair in filter(None, result.query.split("&")))
@ -182,8 +175,7 @@ def _env_rendezvous_handler(url, timeout=default_pg_timeout, **kwargs):
# If this configuration is invalidated, there is nothing we can do about it
raise RuntimeError("Unable to perform rerendezvous using env:// method")
if sys.platform != 'win32':
register_rendezvous_handler("tcp", _tcp_rendezvous_handler)
register_rendezvous_handler("env", _env_rendezvous_handler)
register_rendezvous_handler("file", _file_rendezvous_handler)
register_rendezvous_handler("tcp", _tcp_rendezvous_handler)
register_rendezvous_handler("env", _env_rendezvous_handler)