You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@tvm.apache.org by tq...@apache.org on 2020/07/07 15:18:46 UTC

[incubator-tvm] branch master updated: Fix tune_relay_cuda.py (#6001)

This is an automated email from the ASF dual-hosted git repository.

tqchen pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-tvm.git


The following commit(s) were added to refs/heads/master by this push:
     new 723ed59  Fix tune_relay_cuda.py (#6001)
723ed59 is described below

commit 723ed593f52e3b247c27c5251b64722fc0e00765
Author: Lianmin Zheng <li...@gmail.com>
AuthorDate: Tue Jul 7 08:18:31 2020 -0700

    Fix tune_relay_cuda.py (#6001)
---
 tutorials/autotvm/tune_relay_cuda.py | 6 +-----
 1 file changed, 1 insertion(+), 5 deletions(-)

diff --git a/tutorials/autotvm/tune_relay_cuda.py b/tutorials/autotvm/tune_relay_cuda.py
index a6fe45b..6a995af 100644
--- a/tutorials/autotvm/tune_relay_cuda.py
+++ b/tutorials/autotvm/tune_relay_cuda.py
@@ -128,11 +128,7 @@ tuning_option = {
 
     'measure_option': autotvm.measure_option(
         builder=autotvm.LocalBuilder(timeout=10),
-        #runner=autotvm.LocalRunner(number=20, repeat=3, timeout=4, min_repeat_ms=150),
-        runner=autotvm.RPCRunner(
-            '1080ti',  # change the device key to your key
-            '0.0.0.0', 9190,
-            number=20, repeat=3, timeout=4, min_repeat_ms=150)
+        runner=autotvm.LocalRunner(number=20, repeat=3, timeout=4, min_repeat_ms=150),
     ),
 }