diff --git a/python/tvm/auto_scheduler/testing/tune_onnx.py b/python/tvm/auto_scheduler/testing/tune_onnx.py index 658c5ae4e4231..7e955601dbd5b 100644 --- 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 index c785da2d391fe..f7ad2238c5295 100644 --- 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 index 3580154ccf53d..0c52010184fbd 100644 --- 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 index df568d5e899f6..c295b07a60e08 100644 --- 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, )