Skip to content

GitLab

Unverified Commit 996a9364 authored by Andrei Ivanov's avatar Andrei Ivanov Committed by GitHub
Browse files

[DistDGL] Fixing bug in `test_dist_objects` (#7179)

parent 3fd6a5d0
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 3 deletions
tests/dist/python/run_dist_objects.py
View file @ 996a9364
...@@ -44,7 +44,6 @@ def run_server( ...@@ -44,7 +44,6 @@ def run_server(
server_count, server_count,
num_clients, num_clients,
shared_mem, shared_mem,
keep_alive=False,
): ):
# server_count = num_servers_per_machine # server_count = num_servers_per_machine
g = dgl.distributed.DistGraphServer( g = dgl.distributed.DistGraphServer(
...@@ -55,7 +54,6 @@ def run_server( ...@@ -55,7 +54,6 @@ def run_server(
graph_path + "/{}.json".format(graph_name), graph_path + "/{}.json".format(graph_name),
disable_shared_mem=not shared_mem, disable_shared_mem=not shared_mem,
graph_format=["csc", "coo"], graph_format=["csc", "coo"],
keep_alive=keep_alive,
) )
print("start server", server_id) print("start server", server_id)
g.start() g.start()
...@@ -774,7 +772,6 @@ if mode == "server": ...@@ -774,7 +772,6 @@ if mode == "server":
server_count=num_servers_per_machine, server_count=num_servers_per_machine,
num_clients=num_part * num_client_per_machine, num_clients=num_part * num_client_per_machine,
shared_mem=shared_mem, shared_mem=shared_mem,
keep_alive=False,
) )
elif mode == "client": elif mode == "client":
os.environ["DGL_NUM_SERVER"] = str(num_servers_per_machine) os.environ["DGL_NUM_SERVER"] = str(num_servers_per_machine)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment