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()
