From 8056165eea6b55aa27b6bbed741f58dc3a6e639b Mon Sep 17 00:00:00 2001 From: Tianqi Chen Date: Thu, 2 Aug 2018 21:47:04 -0700 Subject: [PATCH] Fix RPC (#1542) --- nnvm/tests/python/compiler/test_rpc_exec.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/nnvm/tests/python/compiler/test_rpc_exec.py b/nnvm/tests/python/compiler/test_rpc_exec.py index 129a0a425fbc..111ba724e196 100644 --- a/nnvm/tests/python/compiler/test_rpc_exec.py +++ b/nnvm/tests/python/compiler/test_rpc_exec.py @@ -4,12 +4,13 @@ import nnvm.symbol as sym import nnvm.compiler import numpy as np +import time def test_rpc_executor(): host = "localhost" - port = 9120 - server = rpc.Server(host, port) - + port = 9021 + server = rpc.Server(host, port, use_popen=True) + time.sleep(1) x = sym.Variable("x") y = sym.Variable("y") z = sym.exp(y + x) @@ -22,7 +23,7 @@ def test_rpc_executor(): graph, lib, _ = nnvm.compiler.build(z, "llvm", shape_dict) # save module lib.save(lib_name) - remote = rpc.connect(server.host, server.port) + remote = rpc.connect(host, port) remote.upload(lib_name) ctx = remote.cpu(0) # load remote