Skip to content

Commit

Permalink
Fix rpc testcase (#1538)
Browse files Browse the repository at this point in the history
  • Loading branch information
tqchen authored Aug 3, 2018
1 parent 0737fe1 commit af0b73f
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions nnvm/tests/python/compiler/test_rpc_exec.py
Original file line number Diff line number Diff line change
Expand Up @@ -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")
Expand All @@ -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
Expand Down

0 comments on commit af0b73f

Please sign in to comment.