Skip to content

T
tic
Commit 0f431fd6 by Tianqi Chen Committed by GitHub
Options

Fix rpc testcase (#1538)

parent be718ba1
Showing with 3 additions and 3 deletions
nnvm/tests/python/compiler/test_rpc_exec.py
...@@ -7,8 +7,8 @@ import numpy as np ...@@ -7,8 +7,8 @@ import numpy as np
def test_rpc_executor(): def test_rpc_executor():
host = "localhost" host = "localhost"
port = 9711 port = 9120
server = rpc.Server(host, port, use_popen=True) server = rpc.Server(host, port)
x = sym.Variable("x") x = sym.Variable("x")
y = sym.Variable("y") y = sym.Variable("y")
...@@ -22,7 +22,7 @@ def test_rpc_executor(): ...@@ -22,7 +22,7 @@ def test_rpc_executor():
graph, lib, _ = nnvm.compiler.build(z, "llvm", shape_dict) graph, lib, _ = nnvm.compiler.build(z, "llvm", shape_dict)
# save module # save module
lib.save(lib_name) lib.save(lib_name)
remote = rpc.connect(host, port) remote = rpc.connect(server.host, server.port)
remote.upload(lib_name) remote.upload(lib_name)
ctx = remote.cpu(0) ctx = remote.cpu(0)
# load remote # load remote
......
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