Home
last modified time | relevance | path

Searched refs:tcp_store (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/distributed/
H A Drendezvous.py185 tcp_store = TCPStore(hostname, port, world_size, False, timeout)
186 return PrefixStore(f"/worker/attempt_{attempt}", tcp_store)
/aosp_15_r20/external/pytorch/test/distributed/elastic/rendezvous/
H A Ddynamic_rendezvous_test.py1801 tcp_store = Mock(spec=dist.TCPStore)
1804 type(tcp_store).host = PropertyMock(return_value=expected_addr)
1805 type(tcp_store).port = PropertyMock(return_value=expected_port)
1807 self._store = tcp_store
1824 "torch.rendezvous.dummy_run_id.0", tcp_store
1842 tcp_store = Mock(spec=dist.TCPStore)
1844 self._store = tcp_store
/aosp_15_r20/external/pytorch/test/distributed/
H A Dtest_store.py240 tcp_store = dist.TCPStore(
245 for store in [tcp_store, hash_store, file_store]:
877 tcp_store = create_tcp_store(use_libuv=False) variable in TestMultiThreadedWait
924 self._test_wait_tcp_store(self.tcp_store)