Home
last modified time | relevance | path

Searched refs:rpc_backend_options (Results 1 – 18 of 18) sorted by relevance

/aosp_15_r20/external/pytorch/torch/distributed/rpc/
H A D__init__.py89 rpc_backend_options=None, argument
125 if rpc_backend_options is not None and not isinstance(
126 rpc_backend_options, RpcBackendOptions
133 if backend is None and rpc_backend_options is not None:
136 rpc_backend_options,
162 if rpc_backend_options is None:
164 rpc_backend_options = backend_registry.construct_rpc_backend_options(
172 store = dist._create_store_from_options(rpc_backend_options, rank)
179 rpc_backend_options.init_method, rank=rank, world_size=world_size
183 store.set_timeout(timedelta(seconds=rpc_backend_options.rpc_timeout))
[all …]
H A Dbackend_registry.py341 store, name, rank, world_size, rpc_backend_options argument
348 if not isinstance(rpc_backend_options, TensorPipeRpcBackendOptions):
366 rpc_backend_options.device_maps,
367 rpc_backend_options.devices,
385 rpc_backend_options,
396 api._all_gather(None, timeout=rpc_backend_options.rpc_timeout)
412 rpc_backend_options,
/aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/rpc/
H A Drpc_test.py124 def _stub_init_rpc_backend_handler(store, name, rank, world_size, rpc_backend_options): argument
771 rpc_backend_options=self.rpc_backend_options,
859 rpc_backend_options=self.rpc_backend_options,
881 rpc_backend_options=self.rpc_backend_options,
1377 rpc_backend_options=self.rpc_backend_options,
1394 rpc_backend_options=self.rpc_backend_options,
1405 rpc_backend_options=self.rpc_backend_options,
1415 rpc_backend_options=self.rpc_backend_options,
1436 rpc_backend_options=self.rpc_backend_options,
1585 rpc_backend_options=self.rpc_backend_options,
[all …]
H A Dfaulty_agent_rpc_test.py50 self.assertEqual(self.rpc_backend_options.num_worker_threads, 8)
51 self.assertEqual(self.rpc_backend_options.num_fail_sends, 3)
52 self.assertEqual(len(self.rpc_backend_options.messages_to_fail), 4)
53 self.assertEqual(len(self.rpc_backend_options.messages_to_delay), 2)
54 … self.assertEqual(self.rpc_backend_options.rpc_timeout, rpc.constants.DEFAULT_RPC_TIMEOUT_SEC)
60 set(self.rpc_backend_options.messages_to_fail),
65 self.assertEqual(len(self.rpc_backend_options.messages_to_fail), 0)
69 self.assertEqual(self.rpc_backend_options.messages_to_delay, {"SCRIPT_CALL": 1.5})
H A Dtensorpipe_rpc_agent_test_fixture.py20 def rpc_backend_options(self): member in TensorPipeRpcAgentTestFixture
H A Drpc_agent_test_fixture.py37 def rpc_backend_options(self): member in RpcAgentTestFixture
H A Dfaulty_rpc_agent_test_fixture.py37 def rpc_backend_options(self): member in FaultyRpcAgentTestFixture
H A Ddist_autograd_test.py2635 self.assertEqual(self.rpc_backend_options.num_worker_threads, 8)
2636 self.assertEqual(self.rpc_backend_options.num_fail_sends, 3)
2637 self.assertEqual(len(self.rpc_backend_options.messages_to_fail), 4)
2657 options = self.rpc_backend_options
2668 rpc_backend_options=options,
2700 options = self.rpc_backend_options
2711 rpc_backend_options=options,
2738 options = self.rpc_backend_options
2747 rpc_backend_options=options,
/aosp_15_r20/external/pytorch/benchmarks/distributed/rpc/parameter_server/
H A Dlauncher.py238 def run_master(rank, data, args, extra_configs, rpc_backend_options): argument
255 rpc_backend_options=rpc_backend_options,
298 rpc_backend_options = TensorPipeRpcBackendOptions(rpc_timeout=args.rpc_timeout)
301 run_master(rank, data, args, config, rpc_backend_options)
308 rpc_backend_options=rpc_backend_options,
315 rpc_backend_options.set_device_map(server_name, {rank: server_rank})
321 rpc_backend_options=rpc_backend_options,
/aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/_shard/sharded_tensor/
H A D__init__.py39 rpc_backend_options = rpc.TensorPipeRpcBackendOptions(_transports=tp_transports())
40 rpc_backend_options.init_method = f"file://{self.file_name}"
42 rpc_backend_options.set_device_map(
50 rpc_backend_options=rpc_backend_options,
/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Ddist_utils.py66 rpc_backend_options = self.rpc_backend_options
70 rpc_backend_options.rpc_timeout = rpc.constants.DEFAULT_RPC_TIMEOUT_SEC * 5
78 rpc_backend_options=rpc_backend_options,
/aosp_15_r20/external/pytorch/torch/distributed/benchmarks/
H A Dbenchmark_ddp_rpc.py211 rpc_backend_options = TensorPipeRpcBackendOptions()
212 rpc_backend_options.init_method = "tcp://localhost:29500"
275 rpc_backend_options=rpc_backend_options,
286 rpc_backend_options=rpc_backend_options,
/aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/rpc/examples/
H A Dparameter_server_test.py132 rpc_backend_options=self.rpc_backend_options,
140 rpc_backend_options=self.rpc_backend_options,
H A Dreinforcement_learning_rpc_test.py244 rpc_backend_options=self.rpc_backend_options,
259 rpc_backend_options=self.rpc_backend_options,
/aosp_15_r20/external/pytorch/torch/distributed/rpc/_testing/
H A Dfaulty_agent_backend_registry.py30 store, name, rank, world_size, rpc_backend_options argument
39 if not isinstance(rpc_backend_options, FaultyTensorPipeRpcBackendOptions):
49 rpc_backend_options,
/aosp_15_r20/external/pytorch/test/distributed/
H A Dtest_store.py194 rpc_backend_options = rpc.TensorPipeRpcBackendOptions()
195 rpc_backend_options.init_method = f"file://{file.name}"
196 rpc_backend_options._transports = tp_transports()
198 "worker", rank=0, world_size=1, rpc_backend_options=rpc_backend_options
366 rpc_backend_options=backend_opts,
/aosp_15_r20/external/pytorch/test/distributed/_shard/sharded_tensor/
H A Dtest_sharded_tensor.py1101 rpc_backend_options = rpc.TensorPipeRpcBackendOptions(
1104 rpc_backend_options.init_method = f"file://{self.file_name}"
1110 rpc_backend_options=rpc_backend_options,
/aosp_15_r20/external/pytorch/docs/source/
H A Drpc.rst181 >>> rpc_backend_options=rpc.TensorPipeRpcBackendOptions(