Skip to content

Commit

Permalink
[autotvm] fix apache#2617
Browse files Browse the repository at this point in the history
  • Loading branch information
Bing Xu committed Feb 18, 2019
1 parent fef7282 commit 2229d59
Show file tree
Hide file tree
Showing 2 changed files with 6 additions and 6 deletions.
4 changes: 2 additions & 2 deletions python/tvm/autotvm/measure/measure_methods.py
Original file line number Diff line number Diff line change
Expand Up @@ -313,9 +313,9 @@ def set_task(self, task):
from ...rpc.tracker import Tracker
from ...rpc.server import Server

tracker = Tracker('localhost', port=9000, port_end=10000, silent=True)
tracker = Tracker('0.0.0.0', port=9000, port_end=10000, silent=True)
device_key = '$local$device$%d' % tracker.port
server = Server('localhost', port=9000, port_end=10000,
server = Server('0.0.0.0', port=9000, port_end=10000,
key=device_key,
use_popen=True, silent=True,
tracker_addr=(tracker.host, tracker.port))
Expand Down
8 changes: 4 additions & 4 deletions tutorials/autotvm/tune_relay_cuda.py
Original file line number Diff line number Diff line change
Expand Up @@ -112,7 +112,7 @@ def get_network(name, batch_size):
#runner=autotvm.LocalRunner(number=20, repeat=3, timeout=4, min_repeat_ms=150),
runner=autotvm.RPCRunner(
'1080ti', # change the device key to your key
'localhost', 9190,
'0.0.0.0', 9190,
number=20, repeat=3, timeout=4, min_repeat_ms=150)
),
}
Expand Down Expand Up @@ -327,13 +327,13 @@ def tune_and_evaluate(tuning_opt):
#
# .. code-block:: bash
#
# python -m tvm.exec.rpc_server --tracker=localhost:9190 --key=1080ti
# python -m tvm.exec.rpc_server --tracker=0.0.0.0:9190 --key=1080ti
#
# After registering devices, we can confirm it by querying rpc_tracker
#
# .. code-block:: bash
#
# python -m tvm.exec.query_rpc_tracker --host=localhost --port=9190
# python -m tvm.exec.query_rpc_tracker --host=0.0.0.0 --port=9190
#
# For example, if we have four 1080ti, two titanx and one gfx900, the output can be
#
Expand Down Expand Up @@ -362,7 +362,7 @@ def tune_and_evaluate(tuning_opt):
builder=autotvm.LocalBuilder(timeout=10),
runner=autotvm.RPCRunner(
'1080ti', # change the device key to your key
'localhost', 9190,
'0.0.0.0', 9190,
number=20, repeat=3, timeout=4, min_repeat_ms=150),
),
}

0 comments on commit 2229d59

Please sign in to comment.