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