Commit 7c050e42 by Tianqi Chen Committed by GitHub

Fix RPC (#1542)

parent 0f431fd6
...@@ -4,12 +4,13 @@ from tvm.contrib import util, graph_runtime ...@@ -4,12 +4,13 @@ from tvm.contrib import util, graph_runtime
import nnvm.symbol as sym import nnvm.symbol as sym
import nnvm.compiler import nnvm.compiler
import numpy as np import numpy as np
import time
def test_rpc_executor(): def test_rpc_executor():
host = "localhost" host = "localhost"
port = 9120 port = 9021
server = rpc.Server(host, port) server = rpc.Server(host, port, use_popen=True)
time.sleep(1)
x = sym.Variable("x") x = sym.Variable("x")
y = sym.Variable("y") y = sym.Variable("y")
z = sym.exp(y + x) z = sym.exp(y + x)
...@@ -22,7 +23,7 @@ def test_rpc_executor(): ...@@ -22,7 +23,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(server.host, server.port) remote = rpc.connect(host, 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