def setUp(self): self.rank = self.MAIN_PROCESS_RANK self.file = tempfile.NamedTemporaryFile() self.port = common.find_free_port() self.processes = [ self._spawn_process(rank) for rank in range(int(self.world_size)) ]
def setUp(self): # Adding this hack until we fix the FileStore to delete its # content at the end global INIT_METHOD if INIT_METHOD.startswith("file://"): _, filename = tempfile.mkstemp(prefix=FOLDER) INIT_METHOD = "file://{}".format(filename) if INIT_METHOD.startswith("env://"): port = common.find_free_port() os.environ["MASTER_PORT"] = str(port) self.processes = [] self.rank = self.MANAGER_PROCESS_RANK Barrier.init() for rank in range(int(WORLD_SIZE)): self.processes.append(self._spawn_process(rank))
def test_nominal(self): addr = 'localhost' port = common.find_free_port() url = 'tcp://%s:%d?world_size=%d' % (addr, port, 2) gen0 = c10d.rendezvous(url + "&rank=0") store0, rank0, size0 = next(gen0) self.assertEqual(0, rank0) self.assertEqual(2, size0) gen1 = c10d.rendezvous(url + "&rank=1") store1, rank1, size1 = next(gen1) self.assertEqual(1, rank1) self.assertEqual(2, size1) # Set value on both stores store0.set("key0", "value0") store1.set("key1", "value1") # Cross check with get self.assertEqual(b"value0", store1.get("key0")) self.assertEqual(b"value1", store0.get("key1"))
def setUp(self): addr = 'localhost' port = common.find_free_port() self.tcpstore = c10d.TCPStore(addr, port, True) self.prefix = "test_prefix" self.tcpstore.set_timeout(timedelta(seconds=300))
def _create_store(self): addr = 'localhost' port = common.find_free_port() store = c10d.TCPStore(addr, port, True) store.set_timeout(timedelta(seconds=300)) return store
def setUp(self): addr = 'localhost' port = common.find_free_port() self.tcpstore = c10d.TCPStore(addr, port, True) self.prefix = "test_prefix"
def _create_store(self): addr = 'localhost' port = common.find_free_port() return c10d.TCPStore(addr, port, True)