def test_default_comms_no_exist(): cluster = LocalCUDACluster(threads_per_worker=1) client = Client(cluster) cb = default_comms() assert cb is not None cb2 = default_comms() assert cb.sessionId == cb2.sessionId client.close() cluster.close()
def test_default_comms_no_exist(client): try: cb = default_comms() assert cb is not None cb2 = default_comms() assert cb.sessionId == cb2.sessionId finally: cb.destroy()
def test_default_comms_no_exist(cluster): client = Client(cluster) try: cb = default_comms() assert cb is not None cb2 = default_comms() assert cb.sessionId == cb2.sessionId finally: cb.destroy() client.close()
def test_send_recv(n_trials): cluster = LocalCUDACluster(threads_per_worker=1) client = Client(cluster) cb = CommsContext(comms_p2p=True) cb.init() cb = default_comms() start = time.time() dfs = [client.submit(func_test_send_recv, cb.sessionId, n_trials, random.random(), workers=[w]) for wid, w in zip(range(len(cb.worker_addresses)), cb.worker_addresses)] wait(dfs) print("Time: " + str(time.time() - start)) result = list(map(lambda x: x.result(), dfs)) print(str(result)) assert(result) cb.destroy() client.close() cluster.close()
def test_default_comms(client): try: cb = CommsContext(comms_p2p=True, client=client) cb.init() comms = default_comms() assert (cb.sessionId == comms.sessionId) finally: comms.destroy()
def test_default_comms(): cluster = LocalCUDACluster(threads_per_worker=1) client = Client(cluster) cb = CommsContext(comms_p2p=True, client=client) cb.init() comms = default_comms() assert(cb.sessionId == comms.sessionId) comms.destroy() client.close() cluster.close()