OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +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
from torch.testing._internal.common_distributed import (
skip_if_win32,
create_tcp_store
create_tcp_store,
tp_transports
)
from torch.testing._internal.common_utils import (
TestCase,
@ -145,6 +146,7 @@ class FileStoreTest(TestCase, StoreTestBase):
# Init RPC using file
rpc_backend_options = rpc.TensorPipeRpcBackendOptions()
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)
# Init PG using file
@ -258,7 +260,8 @@ class TCPStoreTest(TestCase, StoreTestBase):
)
backend_opts = rpc.TensorPipeRpcBackendOptions(
init_method=f"tcp://{addr}:{port}"
init_method=f"tcp://{addr}:{port}",
_transports=tp_transports()
)
rpc.init_rpc(
name="worker0",