Use vm in rpc run module.

apache · Jun 23, 2022 · 7de8bce · 7de8bce
1 parent 3889ec0
commit 7de8bce
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 0 deletions.
diff --git a/python/tvm/auto_scheduler/testing/tune_onnx.py b/python/tvm/auto_scheduler/testing/tune_onnx.py
@@ -234,6 +234,7 @@ def main():
             dev_type=ARGS.target.kind.name,
             args=input_data,
             continuation=f_timer_vm,
+            use_vm=True,
         )
 
 

diff --git a/python/tvm/auto_scheduler/testing/tune_relay.py b/python/tvm/auto_scheduler/testing/tune_relay.py
@@ -234,6 +234,7 @@ def main():
             dev_type=ARGS.target.kind.name,
             args=input_data,
             continuation=f_timer_vm,
+            use_vm=True,
         )
 
 

diff --git a/python/tvm/meta_schedule/testing/tune_onnx.py b/python/tvm/meta_schedule/testing/tune_onnx.py
@@ -191,6 +191,7 @@ def main():
             dev_type=ARGS.target.kind.name,
             args=input_data,
             continuation=f_timer_vm,
+            use_vm=True,
         )
 
 

diff --git a/python/tvm/meta_schedule/testing/tune_relay.py b/python/tvm/meta_schedule/testing/tune_relay.py
@@ -192,6 +192,7 @@ def main():
             dev_type=ARGS.target.kind.name,
             args=input_data,
             continuation=f_timer_vm,
+            use_vm=True,
         )