From aa8aa298f49c8d92bc42b40d4acc9389c44b62d2 Mon Sep 17 00:00:00 2001 From: tqchen Date: Thu, 2 Aug 2018 19:12:08 -0700 Subject: [PATCH] Fix rpc testcase --- nnvm/tests/python/compiler/test_rpc_exec.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/nnvm/tests/python/compiler/test_rpc_exec.py b/nnvm/tests/python/compiler/test_rpc_exec.py index 614fe08781ab..129a0a425fbc 100644 --- a/nnvm/tests/python/compiler/test_rpc_exec.py +++ b/nnvm/tests/python/compiler/test_rpc_exec.py @@ -7,8 +7,8 @@ def test_rpc_executor(): host = "localhost" - port = 9711 - server = rpc.Server(host, port, use_popen=True) + port = 9120 + server = rpc.Server(host, port) x = sym.Variable("x") y = sym.Variable("y") @@ -22,7 +22,7 @@ def test_rpc_executor(): graph, lib, _ = nnvm.compiler.build(z, "llvm", shape_dict) # save module lib.save(lib_name) - remote = rpc.connect(host, port) + remote = rpc.connect(server.host, server.port) remote.upload(lib_name) ctx = remote.cpu(0) # load remote