def run_server(graph_name,
server_id,
server_count,
num_clients,
shared_mem,
keep_alive=False):
g = DistGraphServer(server_id,
"kv_ip_config.txt",
server_count,
num_clients,
'/tmp/dist_graph/{}.json'.format(graph_name),
disable_shared_mem=not shared_mem,
graph_format=['csc', 'coo'],
keep_alive=keep_alive)
print('start server', server_id)
# verify dtype of underlying graph
cg = g.client_g
for k, dtype in dgl.distributed.dist_graph.FIELD_DICT.items():
if k in cg.ndata:
assert F.dtype(
cg.ndata[k]
) == dtype, "Data type of {} in ndata should be {}.".format(
k, dtype)
if k in cg.edata:
assert F.dtype(
cg.edata[k]
) == dtype, "Data type of {} in edata should be {}.".format(
k, dtype)
g.start()
def run_server(graph_name, server_id, server_count, num_clients, shared_mem):
g = DistGraphServer(server_id, "kv_ip_config.txt", server_count, num_clients,
'/tmp/dist_graph/{}.json'.format(graph_name),
disable_shared_mem=not shared_mem,
graph_format=['csc', 'coo'])
print('start server', server_id)
g.start()
def start_server(rank, tmpdir, disable_shared_mem, graph_name):
g = DistGraphServer(rank,
"rpc_ip_config.txt",
1,
tmpdir / (graph_name + '.json'),
disable_shared_mem=disable_shared_mem)
g.start()
def start_server(rank, tmpdir, disable_shared_mem, num_clients):
import dgl
print('server: #clients=' + str(num_clients))
g = DistGraphServer(rank, "mp_ip_config.txt", 1, num_clients,
tmpdir / 'test_sampling.json', disable_shared_mem=disable_shared_mem,
graph_format=['csc', 'coo'])
g.start()
def start_server(rank, tmpdir):
import dgl
g = DistGraphServer(rank,
"rpc_sampling_ip_config.txt",
1,
"test_sampling",
tmpdir / 'test_sampling.json',
disable_shared_mem=True)
g.start()
def run_server(graph_name, server_id, num_clients, shared_mem, cond_v, shared_v):
g = DistGraphServer(server_id, "kv_ip_config.txt", num_clients, graph_name,
'/tmp/dist_graph/{}.json'.format(graph_name),
disable_shared_mem=not shared_mem)
print('start server', server_id)
cond_v.acquire()
shared_v.value += 1;
cond_v.notify()
cond_v.release()
g.start()
def start_server(rank,
tmpdir,
disable_shared_mem,
graph_name,
graph_format=['csc', 'coo']):
g = DistGraphServer(rank,
"rpc_ip_config.txt",
1,
1,
tmpdir / (graph_name + '.json'),
disable_shared_mem=disable_shared_mem,
graph_format=graph_format)
g.start()
def run_server(graph_name, server_id, num_clients, barrier):
g = DistGraphServer(server_id, server_namebook, num_clients, graph_name,
'/tmp/{}.json'.format(graph_name))
barrier.wait()
print('start server', server_id)
g.start()
def run_server(graph_name, server_id, num_clients, barrier):
g = DistGraphServer(server_id, "kv_ip_config.txt", num_clients, graph_name,
'/tmp/dist_graph/{}.json'.format(graph_name))
barrier.wait()
print('start server', server_id)
g.start()