diff --git a/test/distributed/test_store.py b/test/distributed/test_store.py index 05a220c9fea..2f980858513 100644 --- a/test/distributed/test_store.py +++ b/test/distributed/test_store.py @@ -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",