OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 12:21:27 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

[c10d] Apply EFA workaround to Store tests. (#100382)

Browse Source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/100382
Approved by: https://github.com/fduwjj
This commit is contained in:
Rodrigo Kumpera 2023-05-05 20:18:35 +00:00 committed by PyTorch MergeBot
parent 266c84e3ab
commit 00d4890218
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
test/distributed/test_store.py
View File

@ -20,7 +20,8 @@ if not dist.is_available():
import torch.testing._internal.common_utils as common import torch.testing._internal.common_utils as common
from torch.testing._internal.common_distributed import ( from torch.testing._internal.common_distributed import (
skip_if_win32, skip_if_win32,
create_tcp_store create_tcp_store,
tp_transports
) )
from torch.testing._internal.common_utils import ( from torch.testing._internal.common_utils import (
TestCase, TestCase,
@ -145,6 +146,7 @@ class FileStoreTest(TestCase, StoreTestBase):
# Init RPC using file # Init RPC using file
rpc_backend_options = rpc.TensorPipeRpcBackendOptions() rpc_backend_options = rpc.TensorPipeRpcBackendOptions()
rpc_backend_options.init_method = f"file://{file.name}" rpc_backend_options.init_method = f"file://{file.name}"
rpc_backend_options._transports = tp_transports()
rpc.init_rpc("worker", rank=0, world_size=1, rpc_backend_options=rpc_backend_options) rpc.init_rpc("worker", rank=0, world_size=1, rpc_backend_options=rpc_backend_options)
# Init PG using file # Init PG using file
@ -258,7 +260,8 @@ class TCPStoreTest(TestCase, StoreTestBase):
) )
backend_opts = rpc.TensorPipeRpcBackendOptions( backend_opts = rpc.TensorPipeRpcBackendOptions(
init_method=f"tcp://{addr}:{port}" init_method=f"tcp://{addr}:{port}",
_transports=tp_transports()
) )
rpc.init_rpc( rpc.init_rpc(
name="worker0", name="worker0",