Skip to content

GitLab

"src/git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "00eca4b8876aef281f4cefee6adde60239fa4938"
Unverified Commit 51651eca authored by Jinjing Zhou's avatar Jinjing Zhou Committed by GitHub
Browse files

Fix test failure in multithread rpc test (#3660)

parent e4cb4a37
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 2 deletions
tests/distributed/test_rpc.py
View file @ 51651eca
...@@ -219,7 +219,7 @@ def test_multi_thread_rpc(): ...@@ -219,7 +219,7 @@ def test_multi_thread_rpc():
dgl.distributed.send_request(0, req) dgl.distributed.send_request(0, req)
def subthread_call(server_id): def subthread_call(server_id):
req = HelloRequest(STR, INTEGER, TENSOR+ server_id, simple_func) req = HelloRequest(STR, INTEGER, TENSOR, simple_func)
dgl.distributed.send_request(server_id, req) dgl.distributed.send_request(server_id, req)
...@@ -229,8 +229,9 @@ def test_multi_thread_rpc(): ...@@ -229,8 +229,9 @@ def test_multi_thread_rpc():
res0 = dgl.distributed.recv_response() res0 = dgl.distributed.recv_response()
res1 = dgl.distributed.recv_response() res1 = dgl.distributed.recv_response()
# Order is not guaranteed
assert_array_equal(F.asnumpy(res0.tensor), F.asnumpy(TENSOR)) assert_array_equal(F.asnumpy(res0.tensor), F.asnumpy(TENSOR))
assert_array_equal(F.asnumpy(res1.tensor), F.asnumpy(TENSOR+1)) assert_array_equal(F.asnumpy(res1.tensor), F.asnumpy(TENSOR))
dgl.distributed.exit_client() dgl.distributed.exit_client()
start_client_multithread("rpc_ip_config_multithread.txt") start_client_multithread("rpc_ip_config_multithread.txt")
......
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